def retrieve_directory_list(self): get_directory_list_query = DatabaseQueryHelper.GET_DIRECTORY_LIST_QUERY % self.__client_id cursor.execute(get_directory_list_query) directory_list = cursor.fetchall() print("List of directories: ") for val in directory_list: print(val[0])
def retrieve_file_list(self): __dir_name = self.__dir_key if isinstance(self.__dir_key, int): __get_dir_name_by_id = DatabaseQueryHelper.GET_DIR_NAME_BY_ID_QUERY % self.__dir_key cursor.execute(__get_dir_name_by_id) __dir_name = cursor.fetchall()[0][0] __dir_path = os.path.join(DATA_ROOT, __dir_name) print("Files in " + __dir_name + " directory:") print(os.listdir(__dir_path))
def retrieve_files(self): __dir_name = self.__dir_key __file_name = self.__file_key if isinstance(self.__dir_key, int): __get_dir_name_by_id = DatabaseQueryHelper.GET_DIR_NAME_BY_ID_QUERY % self.__dir_key cursor.execute(__get_dir_name_by_id) __dir_name = cursor.fetchall()[0][0] if isinstance(self.__file_key, int): __get_file_name_by_id = DatabaseQueryHelper.GET_FILE_NAME_BY_ID_QUERY % self.__file_key cursor.execute(__get_file_name_by_id) __file_name = cursor.fetchall()[0][0] else: __get_file_id_by_name = DatabaseQueryHelper.GET_FILE_ID_BY_NAME_QUERY % (self.__file_key, self.__client_id) cursor.execute(__get_file_id_by_name) self.__file_key = cursor.fetchall()[0][0] __file_path = (os.path.join(DATA_ROOT, __dir_name, __file_name)) with open(__file_path) as f: print(f.read())
def __show_dir_id_name(self): get_dir_id_name = DatabaseQueryHelper.GET_DIR_ID_NAME_QUERY % self.__client_id cursor.execute(get_dir_id_name) get_dir_id_name_vals = cursor.fetchall() print("Your directory and directory id: ") print("dir_name" + "\t" + "dir_id") print("*" * 8 + "\t" + "*" * 6) for val in get_dir_id_name_vals: print(val[0] + "\t\t" + str(val[1]))
def remove_files(self): self.__show_dir_id_name() __dir_name = self.__dir_key __file_name = self.__file_key if isinstance(self.__dir_key, int): __get_dir_name_by_id = DatabaseQueryHelper.GET_DIR_NAME_BY_ID_QUERY % self.__dir_key cursor.execute(__get_dir_name_by_id) __dir_name = cursor.fetchall()[0][0] if isinstance(self.__file_key, int): __get_file_name_by_id = DatabaseQueryHelper.GET_FILE_NAME_BY_ID_QUERY % self.__file_key cursor.execute(__get_file_name_by_id) __file_name = cursor.fetchall()[0][0] else: __get_file_id_by_name = DatabaseQueryHelper.GET_FILE_ID_BY_NAME_QUERY % (self.__file_key, self.__client_id) cursor.execute(__get_file_id_by_name) self.__file_key = cursor.fetchall()[0][0] __remove_file = DatabaseQueryHelper.REMOVE_FILE_QUERY % self.__file_key cursor.execute(__remove_file) os.remove(os.path.join(DATA_ROOT, __dir_name, __file_name)) print("File removed successfully.")
def insert_db(self): if self.__is_created: insert_query = DatabaseQueryHelper.INSERT_T_DIR_QUERY % ( self.__dir_name, self.__client_id) cursor.execute(insert_query) cursor.execute(DatabaseQueryHelper.GET_DIR_ID_QUERY) dir_id = str(cursor.fetchall()[0][0]) print("Directory created successfully.") print("Your unique directory id is:", dir_id) else: print("A directory with this name already exist.")
def dir_size(self): __total_size = 0 get_directory_list_query = DatabaseQueryHelper.GET_DIRECTORY_LIST_QUERY % self.__client_id cursor.execute(get_directory_list_query) directory_list = cursor.fetchall() for val in directory_list: dir_path = os.path.join(DATA_ROOT, val[0]) for file in os.listdir(dir_path): __total_size += os.path.getsize(os.path.join(dir_path, file)) __total_size /= 1000000000 if __total_size > 1: print("Storage size exceeded.") print("Delete some data before processing.") print("Terminated.") exit()