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()