def create(conn, client, title, parent=None, class_id=None): # Initialize folder object with title folder = gdata.docs.data.Resource(type="folder", title=title) if parent != None: parent = client.GetResourceById(parent) # Use the Client Object to create the folder in the root of their Drive or the collection specified. folder = client.CreateResource(folder, collection=parent) if conn: if parent != None: Database.insert( conn, Database.structure_insert_string( Utilities.clean_title(title), folder.resource_id.text, parent.resource_id.text, class_id ), ) else: Database.insert( conn, Database.structure_insert_string( Utilities.clean_title(title), folder.resource_id.text, parent, class_id ), ) 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)
def create_flat(conn, client, title, root_collection, parent=None, class_id=None, student_id=None): # Initialize folder object with title folder = gdata.docs.data.Resource(type="folder", title=title) if root_collection != None: root_collection = client.GetResourceById(root_collection) # Use the Client Object to create the folder in the root of their Drive or the collection specified. folder = client.CreateResource(folder, collection=root_collection) # On success insert into database Database.insert( conn, Database.structure_insert_string( Utilities.clean_title(title), folder.resource_id.text, parent, class_id, student_id ), ) return folder