Skip to content
Permalink
Browse files
complete gDrive configuration interface
  • Loading branch information
usmannasir committed Jun 13, 2020
1 parent ca6bd24 commit 570fc149baec996b3e7dbe0565be08a9e79b9226
Show file tree
Hide file tree
Showing 7 changed files with 501 additions and 10 deletions.
@@ -64,7 +64,9 @@ def gDrive(self, request = None, userID = None, data = None):
for items in gDriveAccts:
gDriveAcctsList.append(items.name)

return render(request, 'backup/googleDrive.html', {'accounts': gDriveAcctsList})
websitesName = ACLManager.findAllSites(currentACL, userID)

return render(request, 'backup/googleDrive.html', {'accounts': gDriveAcctsList, 'websites': websitesName})
except BaseException as msg:
return HttpResponse(str(msg))

@@ -88,13 +90,172 @@ def gDriveSetup(self, userID = None, request = None):
gD = GDrive(owner=admin, name=gDriveData['name'],auth=json.dumps(gDriveData))
gD.save()


final_json = json.dumps({'status': 1, 'message': 'Successfully saved.'})
return HttpResponse(final_json)
except BaseException as msg:
final_dic = {'status': 0, 'fetchStatus': 0, 'error_message': str(msg)}
final_json = json.dumps(final_dic)
return HttpResponse(final_json)

def fetchgDriveSites(self, request = None, userID = None, data = None):
try:

userID = request.session['userID']
currentACL = ACLManager.loadedACL(userID)
admin = Administrator.objects.get(pk=userID)

data = json.loads(request.body)

selectedAccount = data['selectedAccount']
recordsToShow = int(data['recordsToShow'])
page = int(str(data['page']).strip('\n'))

gD = GDrive.objects.get(name=selectedAccount)

websites = gD.gdrivesites_set.all()

from s3Backups.s3Backups import S3Backups

pagination = S3Backups.getPagination(len(websites), recordsToShow)
endPageNumber, finalPageNumber = S3Backups.recordsPointer(page, recordsToShow)
finalWebsites = websites[finalPageNumber:endPageNumber]

json_data = "["
checker = 0
counter = 0

from plogical.backupSchedule import backupSchedule

for website in finalWebsites:

dic = {
'name': website.domain
}

if checker == 0:
json_data = json_data + json.dumps(dic)
checker = 1
else:
json_data = json_data + ',' + json.dumps(dic)

counter = counter + 1

json_data = json_data + ']'

currently = gD.runTime

data_ret = {'status': 1, 'websites': json_data, 'pagination': pagination, 'currently': currently}
json_data = json.dumps(data_ret)
return HttpResponse(json_data)


except BaseException as msg:
data_ret = {'status': 0, 'error_message': str(msg)}
json_data = json.dumps(data_ret)
return HttpResponse(json_data)

def addSitegDrive(self, request = None, userID = None, data = None):
try:

userID = request.session['userID']
currentACL = ACLManager.loadedACL(userID)
admin = Administrator.objects.get(pk=userID)

data = json.loads(request.body)

selectedAccount = data['selectedAccount']
selectedWebsite = data['selectedWebsite']

gD = GDrive.objects.get(name=selectedAccount)

gdSite = GDriveSites(owner=gD, domain=selectedWebsite)
gdSite.save()

data_ret = {'status': 1}
json_data = json.dumps(data_ret)
return HttpResponse(json_data)

except BaseException as msg:
data_ret = {'status': 0, 'error_message': str(msg)}
json_data = json.dumps(data_ret)
return HttpResponse(json_data)

def deleteAccountgDrive(self, request = None, userID = None, data = None):
try:

userID = request.session['userID']
currentACL = ACLManager.loadedACL(userID)
admin = Administrator.objects.get(pk=userID)

data = json.loads(request.body)

selectedAccount = data['selectedAccount']

gD = GDrive.objects.get(name=selectedAccount)

gD.delete()

data_ret = {'status': 1}
json_data = json.dumps(data_ret)
return HttpResponse(json_data)

except BaseException as msg:
data_ret = {'status': 0, 'error_message': str(msg)}
json_data = json.dumps(data_ret)
return HttpResponse(json_data)

def changeAccountFrequencygDrive(self, request = None, userID = None, data = None):
try:

userID = request.session['userID']
currentACL = ACLManager.loadedACL(userID)
admin = Administrator.objects.get(pk=userID)

data = json.loads(request.body)

selectedAccount = data['selectedAccount']
backupFrequency = data['backupFrequency']

gD = GDrive.objects.get(name=selectedAccount)
gD.runTime = backupFrequency

gD.save()

data_ret = {'status': 1}
json_data = json.dumps(data_ret)
return HttpResponse(json_data)

except BaseException as msg:
data_ret = {'status': 0, 'error_message': str(msg)}
json_data = json.dumps(data_ret)
return HttpResponse(json_data)

def deleteSitegDrive(self, request = None, userID = None, data = None):
try:

userID = request.session['userID']
currentACL = ACLManager.loadedACL(userID)
admin = Administrator.objects.get(pk=userID)

data = json.loads(request.body)

selectedAccount = data['selectedAccount']
website = data['website']

gD = GDrive.objects.get(name=selectedAccount)
gDSite = GDriveSites.objects.get(owner=gD, domain=website)
gDSite.delete()

data_ret = {'status': 1}
json_data = json.dumps(data_ret)
return HttpResponse(json_data)

except BaseException as msg:
data_ret = {'status': 0, 'error_message': str(msg)}
json_data = json.dumps(data_ret)
return HttpResponse(json_data)

def restoreSite(self, request = None, userID = None, data = None):
try:
currentACL = ACLManager.loadedACL(userID)

0 comments on commit 570fc14

Please sign in to comment.