Typing

server/authentication.py

@@ -25,13 +25,11 @@ class Authetication:
 
         if not os.path.isfile(self.CONFIG_DIRECTORY_LOCATION + "\\config.txt") or os.stat(
                 self.CONFIG_DIRECTORY_LOCATION + "\\config.txt").st_size == 0:
-            data = {}
-            data['index'] = 0
-            data['user'] = []
+            data = {'index': 0, 'user': []}
             with open(self.CONFIG_DIRECTORY_LOCATION + "\\config.txt", 'w+') as outfile:
                 json.dump(data, outfile)
 
-    def login(self, username, password):
+    def login(self, username: str, password: str) -> str:
         with open(Authetication.CONFIG_DIRECTORY_LOCATION + '\\config.txt') as json_file:
             data = json.load(json_file)
 
@@ -47,7 +45,7 @@ class Authetication:
         auth_logger.debug("User logged in: " + username)
         return user['homeDir']
 
-    def checkUserExists(self, username):
+    def checkUserExists(self, username: str) -> bool:
         with open(Authetication.CONFIG_DIRECTORY_LOCATION + '\\config.txt') as json_file:
             data = json.load(json_file)
 
@@ -64,7 +62,7 @@ class Authetication:
         shutil.rmtree(self.HOME_DIRECTORY_LOCATION)
         os.mkdir(self.HOME_DIRECTORY_LOCATION)
 
-    def saveUser(self, username, password):
+    def saveUser(self, username: str, password: str) -> bool:
         bytePass = password.encode('utf-8')
         b64pwd = b64encode(SHA256.new(bytePass).digest())
         bcrypt_hash = bcrypt(b64pwd, 12)

server/authentication_test.py

@@ -6,7 +6,7 @@ test_logger = logging.getLogger('TEST ')
 test_logger.root.setLevel(logging.INFO)
 
 
-def testSaveUser(username, password):
+def testSaveUser(username: str, password: str):
     logging.info('STARTING SAVE USER TEST')
     auth = Authetication()
     auth.initConfig()
@@ -26,7 +26,7 @@ def testSaveUser(username, password):
         test_logger.info('TEST 1 ---> Save VALID user :: FAILED')
 
 
-def testAuth(username, password):
+def testAuth(username: str, password: str):
     logging.info('STARTING AUTHENTICATION TEST')
     auth = Authetication()
     auth.initConfig()
@@ -45,7 +45,7 @@ def testAuth(username, password):
         test_logger.info('TEST 2 --> Authentication test with INVALID ::FAILED')
 
 
-def testUserExists(username, password):
+def testUserExists(username: str, password: str):
     logging.info('STARTING USER EXISTS TEST')
     auth = Authetication()
     auth.initConfig()