Fix register userdata bug

Author: wnelson03

keyauth.py

@@ -114,13 +114,12 @@ def register(self, user, password, license, hwid=None):
         }
 
         response = self.__do_request(post_data)
-
         response = encryption.decrypt(response, self.enckey, init_iv)
-
         json = jsond.loads(response)
 
         if json["success"]:
             print("successfully registered")
+            self.__load_user_data(json["info"])
         else:
             print(json["message"])
             sys.exit()
@@ -147,6 +146,8 @@ def upgrade(self, user, license):
 
         if json["success"]:
             print("successfully upgraded user")
+            print("please restart program and login")
+            sys.exit()
         else:
             print(json["message"])
             sys.exit()
@@ -511,4 +512,4 @@ def decrypt(message, enc_key, iv):
             return encryption.decrypt_string(message.encode(), _key.encode(), _iv.encode()).decode()
         except:
             print("Invalid Application Information. Long text is secret short text is ownerid. Name is supposed to be app name not username")
-            sys.exit()
+            sys.exit()

main.py

@@ -29,12 +29,6 @@ def getchecksum():
 	version = "1.0",
 	hash_to_check = getchecksum()
 )
-print ("""
-1.Login
-2.Register
-3.Upgrade
-4.License Key Only
-""")
 print(f"""
 App data:
 Number of users: {keyauthapp.app_data.numUsers}
@@ -43,11 +37,14 @@ def getchecksum():
 Application Version: {keyauthapp.app_data.app_ver}
 Customer panel link: {keyauthapp.app_data.customer_panel}
 """)
-sleep(1.5) # rate limit
 print(f"Current Session Validation Status: {keyauthapp.check()}")
-sleep(1.5) # rate limit
 print(f"Blacklisted? : {keyauthapp.checkblacklist()}") # check if blacklisted, you can edit this and make it exit the program if blacklisted
-sleep(1.5)
+print ("""
+1.Login
+2.Register
+3.Upgrade
+4.License Key Only
+""")
 ans=input("Select Option: ") 
 if ans=="1": 
 	user = input('Provide username: ')
@@ -62,6 +59,7 @@ def getchecksum():
 	user = input('Provide username: ')
 	license = input('Provide License: ')
 	keyauthapp.upgrade(user,license)
+    
 elif ans=="4":
 	key = input('Enter your license: ')
 	keyauthapp.license(key)
@@ -126,4 +124,4 @@ def getchecksum():
 print(f"Current Session Validation Status: {keyauthapp.check()}")
 print("Exiting in 10 secs....")
 sleep(10)
-exit(0)
+exit(0)