diff --git a/gpsEdit/main.py b/gpsEdit/main.py index dc42c4d..82a088d 100644 --- a/gpsEdit/main.py +++ b/gpsEdit/main.py @@ -5,42 +5,41 @@ import os.path import shutil media_dir = "//sr3//home//Temp//" -file_remove = "-edited" -edited_list = [] -vanilla_list = [] +media_list = [] +vanilla_media_list = [] file_move_cnt = 0 -def get_list_of_files(): - global edited_list - edited_list = glob.glob(media_dir + "*", recursive=True) +def build_media(): + global media_list + media_list = glob.glob(media_dir + "*", recursive=True) -def find_vanilla_files(): - global vanilla_list, edited_list, file_remove - for x in edited_list: - c_file = str(x).replace(file_remove, "") +def build_vanilla_media(): + global vanilla_media_list, media_list + for x in media_list: + c_file = str(x).replace("-edited", "") if os.path.isfile(c_file): - if not vanilla_list.__contains__(c_file): - vanilla_list.append(c_file) + if not vanilla_media_list.__contains__(c_file): + vanilla_media_list.append(c_file) -def move_vanilla_files(): - global media_dir, vanilla_list, file_move_cnt - del_dir = media_dir + "ToDelete\\" +def move_vanilla_media(): + global media_dir, vanilla_media_list, file_move_cnt + dest_dir = media_dir + "_VanillaMedia\\" - if not os.path.exists(del_dir): - os.mkdir(del_dir) + if not os.path.exists(dest_dir): + os.mkdir(dest_dir) - for x in vanilla_list: - shutil.move(x, del_dir) + for x in vanilla_media_list: + shutil.move(x, dest_dir) file_move_cnt += 1 print("Moving " + str(x)) if __name__ == "__main__": - get_list_of_files() - find_vanilla_files() - move_vanilla_files() + build_media() + build_vanilla_media() + move_vanilla_media() print("Finished moving " + str(file_move_cnt) + " file(s).")