cloudperm/revokeGUI.py at master · deltaray/cloudperm · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
#!/usr/bin/env python3
from __future__ import print_function

from tkinter import *
from tkinter.messagebox import showinfo
from random import *

from cloudperm import *
# from revokeAccess import *
import argparse
parser = argparse.ArgumentParser(parents=[cloudperm_argparser])
args = parser.parse_args()

credentials = get_credentials(args)
http = credentials.authorize(httplib2.Http())
service = discovery.build('drive', 'v2', http=http)

def m_rev(email, files):
    # print("FILES", files)
    files = str(files)
    files = files.strip()
    fileids = files.split(', ')

    email = email.strip()
    accountToRevoke=str(email)
    print(accountToRevoke)
    for id in fileids:
        id = id.strip()
    # print("FIDS", fileids)
    for fileid in fileids:
        title = retrieve_document_title(service, fileid);
        # print("TITLE", title)
        perm_list = retrieve_permissions(service, fileid)
        accountrevoked = False;

        for entry in perm_list:
            # print("permlist", perm_list)
            if type(entry) is dict:
                if 'emailAddress' in entry:
                    if entry['emailAddress'] == accountToRevoke:
                        permIdToRevoke = entry['id']
                        revoke_response = revoke_document_role(service, fileid, permIdToRevoke);

                        pp = pprint.PrettyPrinter(indent=8,depth=6)
                        if revoke_response:
                            pp.pprint(revoke_response);
                        else: # An empty response to
                            accountrevoked = True;
                            # print("Revoked access for " + accountToRevoke + " from document '" + title + "'");
                            print("Access has successfully wbeen revoked for", email, "on", fileid)

        if not accountrevoked:
            # print("WARNING: access was not revoked for " + accountToRevoke + " on file '" + title.encode('utf-8') + "'");ess was not revoked for " + accountToRevoke + " on file '" + title.encode('utf-8') + "'");
            print("ACCOUNT WAS UNABLE TO BE REVOKED")

class Application(Frame):
    def __init__(self, master):
        Frame.__init__(self, master)
        self.grid()
        self.create_widgets()

    def create_widgets(self):
         #creates label
        self.lbl = Label(self, text="Enter the email address to revoke access to:")
        self.lbl.grid(row =  1, column = 0, sticky = W)

        #creates Entry box
        self.ent = Entry(self, width = 15)
        self.ent.grid(row = 2, column = 0, sticky = W)

        ###############################
        self.lbl1 = Label(self, text="Enter a fileID or list of fileIDs seperated by commas:")
        self.lbl1.grid(row =  3, column = 0, sticky = W)

        #creates Entry box
        self.ent1 = Entry(self, width = 50)
        self.ent1.grid(row = 4, column = 0, sticky = W)

        #creates button / when pressed command = self.enter
        self.bttn1 = Button(self, text = "Enter", command = self.enter).grid(row = 7, column = 0, sticky = W)

        #binds keypress
        self.ent.bind('<Return>', self.enter_click)


    def enter(self):
        #gets value out of entry box
        email = self.ent.get()
        files = self.ent1.get()
        self.ent.delete(0, END)
        self.ent1.delete(0, END)
        m_rev(email, files)
        #delets enty in the box


    def enter_click(self, event):
        if event.keysym == "Return" or "Enter":
            self.enter()

#main
root = Tk()
root.title("Revoke Access")
root.geometry("800x300")
root.resizable(width = TRUE, height = TRUE)
app = Application(root)
root.mainloop()