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)
