def folder_not_exists_create(client, conn, folder, template, parent, currentdir_folders): if folder['folder_name'] in currentdir_folders: #folder exists, update the folder_id and return it folder['folder_id'] = currentdir_folders[folder['folder_name']] return folder elif folder['copy']: #this is a folder that needs to be copied from somewhere else if template == "{{STUDENT_ASSIGNMENTS}}": Folder.copy(client, folder['folder_id'], parent) return folder elif template == "{{CLASS_FILES}}": table = "vlacs_class_folders_shared" cols = {'folder_name':{'value':folder['folder_name'],'type':'s'}} class_files = Database.insert_if_not_exists(conn, table, cols) if class_files: cr_folder = Folder.create(False, client, folder['folder_name'], parent) cols = {'folder_id':{'value':cr_folder.resource_id.text,'type':'s'}} wheres = {'folder_name':{'value':folder['folder_name'],'type':'s'}} Database.update(conn, table, cols, wheres) folder['folder_id'] = cr_folder.resource_id.text else: folder['folder_id'] = class_files['folder_id'] Folder.copy(client, folder['folder_id'], parent) return folder else: #folder does not exist, create it and update the folder_id and return it cr_folder = Folder.create(False, client, folder['folder_name'], parent) folder['folder_id'] = cr_folder.resource_id.text return folder
def fix_structure(client, conn): tables_exist = False folder_list = {} exists_list_gd = {} exists_list_db = {} everything_exists = False print "Making sure the database tables exist..." # CHECK FOR DATABASE TABLES # tables_query = Database.get(Database.execute(conn, "SELECT COUNT(*) FROM pg_tables WHERE schemaname='public' AND tablename LIKE 'vlacs_class_folders%'")) if tables_query['count'] > 1: print "Database tables exist." tables_exist = True if not tables_exist: print "Database tables do not exist, creating..." Database.insert(conn, "CREATE TABLE IF NOT EXISTS vlacs_class_folders_structure(id serial, class_id integer, student_id integer, folder_name text, folder_id text, folder_parent text, isactive int DEFAULT 1 NOT NULL)") Database.insert(conn, "CREATE TABLE IF NOT EXISTS vlacs_class_folders_shared(id serial, folder_id text, folder_name text, shared_with text, shared_permission text)") print "Making sure the root folders exist in Google Drive and Database..." # STORE NAMES OF ROOT LEVEL FOLDERS IN LIST # for resource in client.GetAllResources(uri="/feeds/default/private/full/root/contents/-/folder", show_root=True): if resource.GetResourceType() == 'folder': folder_list[resource.title.text] = resource.resource_id.text for folder_name in [config.ROOT_CLASS_FOLDER, config.TEACHER_SHARE_FOLDER, config.STUDENT_SHARE_FOLDER]: # CHECK IF FOLDER EXISTS IN DRIVE # for title, f_id in folder_list.items(): if title == folder_name: print "--- %s exists in Google Drive." % (folder_name) exists_list_gd[folder_name] = True #CHECK IF FOLDER EXISTS IN DATATBASE # cq = Database.get(Database.execute(conn, "SELECT count(*) FROM vlacs_class_folders_structure WHERE folder_name = '%s'" % folder_name)) if cq['count'] > 0: print "--- %s exists in the Database." % (folder_name) exists_list_db[folder_name] = True if (config.ROOT_CLASS_FOLDER in exists_list_db and config.ROOT_CLASS_FOLDER in exists_list_gd and config.TEACHER_SHARE_FOLDER in exists_list_db and config.TEACHER_SHARE_FOLDER in exists_list_gd and config.STUDENT_SHARE_FOLDER in exists_list_db and config.STUDENT_SHARE_FOLDER in exists_list_gd): everything_exists = True if not everything_exists: print "Something is missing..." # COMPARE AND INSERT / CREATE # for folder, config_f in [{'root', config.ROOT_CLASS_FOLDER}, {'teacher', config.TEACHER_SHARE_FOLDER}, {'student', config.STUDENT_SHARE_FOLDER}]: if folder in exists_list_db and folder not in exists_list_gd: print "--- %s folder is in the database, but not Google Drive. Fixing..." % (config_f) f = Folder.create(False, client, config_f) Database.insert(conn, "UPDATE vlacs_class_folders_structure SET folder_id = '%s' WHERE folder_name = '%s'" % (f.resource_id.text, config_f)) elif folder in exists_list_gd and folder not in exists_list_db: print "--- %s folder is in Google Drive but not in the database. Fixing..." % (config_f) Database.insert(conn, Database.structure_insert_string(config_f, folder_list[config_f])) elif folder not in exists_list_db and folder_list not in exists_list_gd: print "--- %s folder is not in Google Drive or the database. Fixing..." % (config_f) f = Folder.create(conn, client, config_f)