client/app.py

@@ -1,15 +1,18 @@
 import tkinter as tk
+
 import requests
 
 FONT = ("Verdana", 12)
+FONT_B = ("Verdana", 12, 'bold')
 FONT_LARGE = ("Verdana", 20)
 URL = "http://localhost:8000/api/authenticate"
-TOKEN = ""
+URL_REGISTER = "http://localhost:8000/api/register"
+URL_PROFILE = "http://localhost:8000/api/profile"
+
 
 class SmartPicasso(tk.Tk):
 
     def __init__(self, *args, **kwargs):
-        
         tk.Tk.__init__(self, *args, **kwargs)
         container = tk.Frame(self)
         self.title('SmartPicasso')
@@ -22,23 +25,24 @@ class SmartPicasso(tk.Tk):
 
         self.frames = {}
 
-        for F in (LoginPage, MainView, RegisterView):
-
+        for F in (LoginView, MainView, RegisterView):
             frame = F(container, self)
 
             self.frames[F] = frame
 
             frame.grid(row=0, column=0, sticky="nsew")
 
-        self.show_frame(LoginPage)
+        self.show_frame(LoginView)
 
-    def show_frame(self, cont):
-
-        frame = self.frames[cont]
+    def show_frame(self, view, token=None):
+        frame = self.frames[view]
         frame.tkraise()
 
+        if token:
+            frame.token = token
 
-class LoginPage(tk.Frame):
+
+class LoginView(tk.Frame):
 
     def __init__(self, parent, controller):
         tk.Frame.__init__(self, parent)
@@ -54,32 +58,34 @@ class LoginPage(tk.Frame):
         label2 = tk.Label(self, text='Password:', font=FONT)
         label2.pack()
 
-        input2 = tk.Entry(self)
+        input2 = tk.Entry(self, show="*")
         input2.pack()
 
-        button = tk.Button(self, text="Login", font=FONT, command=lambda: self.login(controller, input1.get(), input2.get()))
+        button = tk.Button(self, text="Login", font=FONT,
+                           command=lambda: self.login(controller, input1.get(), input2.get()))
         button.pack()
 
         button2 = tk.Button(self, text="Register", font=FONT, command=lambda: controller.show_frame(RegisterView))
         button2.pack()
 
-    def login(self, controller, login, passw,):
+    def login(self, controller, login, password):
         print(login)
-        print(passw)
+        print(password)
         data = {
             "email": str(login),
-            "password": str(passw)
+            "password": str(password)
         }
         resp = requests.post(URL, json=data)
         print(resp)
-        if (resp.status_code==200):
+        if resp.status_code == 200:
             response = resp.json()
-            TOKEN = response['token']
-            controller.show_frame(MainView)
+            token = response['token']
+
+            controller.show_frame(MainView, token)
         else:
             print("bad pass")
-            badPassLabel = tk.Label(self, text='Wrong login/password!', font=FONT)
-            badPassLabel.pack() 
+            bad_pass_label = tk.Label(self, text='Wrong login/password!', font=FONT)
+            bad_pass_label.pack()
         return ()
 
 
@@ -87,11 +93,32 @@ class MainView(tk.Frame):
 
     def __init__(self, parent, controller):
         tk.Frame.__init__(self, parent)
+        self.frames = {}
+        self.token = ''
+        for F in (ProfileView,):
+            frame = F(parent, controller)
+
+            self.frames[F] = frame
+
+            frame.grid(row=0, column=0, sticky="nsew")
         label = tk.Label(self, text="SmartPicasso", font=FONT_LARGE)
         label.pack(pady=10, padx=10)
         label_u = tk.Label(self, text="Main menu", font=FONT)
         label_u.pack(pady=10, padx=10)
 
+        button_profile = tk.Button(self, text="My profile", font=FONT,
+                                   command=lambda: self.show_frame(ProfileView))
+        button_profile.pack()
+
+    def show_frame(self, view):
+        frame = self.frames[view]
+        frame.tkraise()
+
+        if self.token:
+            print(self.token)
+            frame.token = self.token
+            frame.start()
+
 
 class RegisterView(tk.Frame):
 
@@ -102,6 +129,12 @@ class RegisterView(tk.Frame):
         label_u = tk.Label(self, text="Register", font=FONT)
         label_u.pack(pady=10, padx=10)
 
+        label0 = tk.Label(self, text='Email:', font=FONT)
+        label0.pack()
+
+        input0 = tk.Entry(self)
+        input0.pack()
+
         label1 = tk.Label(self, text='Login:', font=FONT)
         label1.pack()
 
@@ -111,17 +144,84 @@ class RegisterView(tk.Frame):
         label2 = tk.Label(self, text='Password:', font=FONT)
         label2.pack()
 
-        input2 = tk.Entry(self)
+        input2 = tk.Entry(self, show="*")
         input2.pack()
 
-        label3 = tk.Label(self, text='Email:', font=FONT)
+        label3 = tk.Label(self, text='First name:', font=FONT)
         label3.pack()
 
         input3 = tk.Entry(self)
         input3.pack()
 
-        button = tk.Button(self, text="Register", font=FONT, command=lambda: controller.show_frame(RegisterView))
-        button.pack()
+        label4 = tk.Label(self, text='Last name:', font=FONT)
+        label4.pack()
+
+        input4 = tk.Entry(self)
+        input4.pack()
+
+        button1 = tk.Button(self, text="Register", font=FONT,
+                            command=lambda: self.register(controller, input0.get(), input1.get(), input2.get(),
+                                                          input3.get(), input4.get()))
+        button1.pack()
+
+        button2 = tk.Button(self, text="Cancel", font=FONT, command=lambda: controller.show_frame(LoginView))
+        button2.pack()
+
+    def register(self, controller, email, login, passw, name, lastname):
+        data = {
+            "email": str(email),
+            "password": str(passw),
+            "profile": {
+                "username": str(login),
+                "first_name": str(name),
+                "last_name": str(lastname)
+            }
+        }
+        print(data)
+        response = requests.post(URL_REGISTER, json=data)
+        print(response)
+        if response.status_code == 201:
+            response = response.json()
+            controller.show_frame(LoginView)
+        else:
+            print("sth wrong")
+            bad_pass_label = tk.Label(self, text='Something went wrong!', font=FONT)
+            bad_pass_label.pack()
+        return ()
+
+
+class ProfileView(tk.Frame):
+
+    def __init__(self, parent, controller):
+        tk.Frame.__init__(self, parent)
+        self.token = ''
+        label = tk.Label(self, text="SmartPicasso", font=FONT_LARGE)
+        label.pack(pady=10, padx=10)
+        label_l1 = tk.Label(self, text="Login:", font=FONT_B)
+        label_l1.pack(pady=10, padx=10)
+        self.label_username = tk.Label(self, text='', font=FONT)
+        self.label_username.pack(pady=10, padx=10)
+        label_n1 = tk.Label(self, text="Name:", font=FONT_B)
+        label_n1.pack(pady=10, padx=10)
+        self.label_first_name = tk.Label(self, text='', font=FONT)
+        self.label_first_name.pack(pady=10, padx=10)
+        label_ln1 = tk.Label(self, text="Last name", font=FONT_B)
+        label_ln1.pack(pady=10, padx=10)
+        self.label_last_name = tk.Label(self, text='', font=FONT)
+        self.label_last_name.pack(pady=10, padx=10)
+
+        button_profile = tk.Button(self, text="Back", font=FONT,
+                                   command=lambda: controller.show_frame(MainView, self.token))
+        button_profile.pack()
+
+    def start(self):
+        headers = {'Authorization': 'Bearer ' + self.token}
+        resp = requests.get(URL_PROFILE, headers=headers)
+        response = resp.json()
+        print(response)
+        self.label_username['text'] = response['profile']['username']
+        self.label_first_name['text'] = response['profile']['first_name']
+        self.label_last_name['text'] = response['profile']['last_name']
 
 
 app = SmartPicasso()

client/requirements.txt

@@ -0,0 +1,2 @@
+tkinter
+requests