Synnefo » snf-ganeti » Ganeti ExtStorage providers

#!/usr/bin/env python

#

# Copyright (C) 2012 Greek Research and Technology Network

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful, but

# WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

# General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA

# 02110-1301, USA.

"""Remove an existing file under a shared directory

This program removes an existing file from a shared directory. It takes

it's input from environment variables. Specifically the following variables

should be present:

 - VOL_NAME: The name of the new Image file

Returns O after successfull creation, 1 on failure

"""

import os

import sys

sys.path.insert(0, "/etc/ganeti/share")

from ganeti import utils

# Default shared directory

SHARED_DIR = "/srv/ganeti/shared-file-storage"

def ReadEnv():

  """Read the mandatory enviromental variables

  """

  name = os.getenv("VOL_NAME")

  if name is None:

    sys.stderr.write('The environment variable VOL_NAME is missing.\n')

    return None

  shared_dir_param = os.getenv("EXTP_SHARED_DIR")

  if shared_dir_param is not None:

    sys.stderr.write('Found shared dir param, value: %s\n' % shared_dir_param)

  return (name, shared_dir_param)

def main():

  sys.stderr.write('Removal started: reading environment...\n')

  # Read environment

  env = ReadEnv()

  if env is None:

    sys.stderr.write('Wrong environment. Aborting\n')

    return 1

  file_name, shared_dir_param = env

  if shared_dir_param is None:

    shared_dir_param = SHARED_DIR

  file_path = utils.PathJoin(shared_dir_param, file_name)

  sys.stderr.write('Read environment successfully: file name: %s\n' % file_path)

  # Remove file

  cmd = ["rm", "-f", "%s" % file_path]

  result = utils.RunCmd(cmd)

  if result.failed:

    sys.stderr.write('Cannot remove Image file %s with `rm -f`: %s - %s\n' %

                     (file_path, result.fail_reason, result.output))

    return 1

  sys.stderr.write('Image file %s removed successfully\n' % file_path)

  return 0

if __name__ == "__main__":

    sys.exit(main())