#!/usr/bin/env python

#------------------------------------------------------------------------------------------
## Copyright 2010 Alexey Petrov
##
## Licensed under the Apache License, Version 2.0 (the "License");
## you may not use this file except in compliance with the License.
## You may obtain a copy of the License at
##
##     http://www.apache.org/licenses/LICENSE-2.0
##
## Unless required by applicable law or agreed to in writing, software
## distributed under the License is distributed on an "AS IS" BASIS,
## WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
## See the License for the specific language governing permissions and
## limitations under the License.


#------------------------------------------------------------------------------------------
"""
Removes whole appointed cloud study or just given files from this study
"""

#------------------------------------------------------------------------------------------
import balloon.common as common
from balloon.common import print_d, print_i, print_e, sh_command
from balloon.common import Timer, WorkerPool, compute_md5

import balloon.amazon as amazon
from balloon.amazon.s3 import TRootObject, TStudyObject, TFileObject, TSeedObject

import sys, os, os.path, uuid, hashlib


#------------------------------------------------------------------------------------------
# Defining utility command-line interface
an_usage_description = "%prog --study-name=<my favorite study>"
an_usage_description += common.add_usage_description()
an_usage_description += amazon.add_usage_description()
an_usage_description += amazon.add_timeout_usage_description()
an_usage_description += amazon.add_threading_usage_description()

from optparse import IndentedHelpFormatter
a_help_formatter = IndentedHelpFormatter( width = 127 )

from optparse import OptionParser
an_option_parser = OptionParser( usage = an_usage_description, version="%prog 0.1", formatter = a_help_formatter )

# Definition of the command line arguments
an_option_parser.add_option( "--study-name",
                            metavar = "< an unique name of the user study >",
                            action = "store",
                            dest = "study_name",
                            default = None )
common.add_parser_options( an_option_parser, True )
amazon.add_parser_options( an_option_parser )
amazon.add_timeout_options( an_option_parser )
amazon.add_threading_parser_options( an_option_parser )


#------------------------------------------------------------------------------------------
# Extracting and verifying command-line arguments

an_options, an_args = an_option_parser.parse_args()

a_study_name = an_options.study_name
if a_study_name == None :
    an_option_parser.error( "Use --study-name option to define proper value\n" )
    pass
print_d( "a_study_name = '%s'\n" % a_study_name )

a_located_files = None
if len( an_args ) == 0 :
    a_located_files = [ raw_input() ]
else :
    a_located_files = an_args
    pass

if a_located_files == None :
    an_option_parser.error( "Define 'located files' to be removed - nothing to be done for now\n" )
    pass

print_d( "a_located_files = %r\n" % a_located_files )

common.extract_options( an_options )

AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY = amazon.extract_options( an_options )

a_number_threads = amazon.extract_threading_options( an_options, an_option_parser )

amazon.extract_timeout_options( an_options, an_option_parser )


print_i( "--------------------------- Looking for study object ----------------------------\n" )
a_root_object = TRootObject.get( AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY )
print_d( "a_root_object = %s\n" % a_root_object )

a_study_object = TStudyObject.get( a_root_object, a_study_name )
print_d( "a_study_object = %s\n" % a_study_object )


print_i( "-------------------------- Removing study files -----------------------------\n" )
for a_located_file in a_located_files :
    print_d( "a_located_file = %s\n" % a_located_file )
    
    a_file_object = TFileObject.get( a_study_object, a_located_file )
    print_d( "a_file_object = %s\n" % a_file_object, 1 )

    a_file_object.delete()
    
    pass


print_i( "-------------------------------------- OK ---------------------------------------\n" )
