Merge pull request #144 from PyBotDevs/release-IsoBank-beta

Release IsoBank registration to all users

Author: notsniped

framework/isobank/authorize.py

@@ -17,16 +17,16 @@ def __init__(self, db_path:str, account_db:str):
             user_data = json.load(f)
 
     def save(self):
-        with open(self.db_path, 'w+') as f: json.dump(accounts, indent=4)
-        with open(self.db_path, 'w+') as f: json.dump(user_data, indent=4)
+        with open(self.db_path, 'w+') as f: json.dump(accounts, f, indent=4)
+        with open(self.account_db, 'w+') as f: json.dump(user_data, f, indent=4)
 
     def register(self, discord_id:int, auth_id:int):
         if disabled: return "[!] IsoBank is currently disabled."
         if discord_id in accounts: return "[!] That user is already registered!"
         user_count = len(accounts.keys())
         new_id = user_count + 1
-        if not auth_id.isdigit(): return "\"auth_id\" is not an integer."
-        if len(auth_id) != 6: return "\"auth_id\" must be passed as a 6-digit number."
+        if not str(auth_id).isdigit(): return "\"auth_id\" is not an integer."
+        if len(str(auth_id)) != 6: return "\"auth_id\" must be passed as a 6-digit number."
         accounts[str(new_id)] = {"discord_ids": [discord_id], "auth_id": auth_id}
         user_data[str(new_id)] = {"deposited": 0}
         self.save()
@@ -37,5 +37,7 @@ def authorize(self, discord_id:int, account_id:int, auth_id:int):
         if disabled: return "[!] IsoBank is currently disabled."
         wacc = accounts[str(account_id)]
         if wacc["auth_id"] != auth_id: return "Incorrect auth ID"
-        if discord_id not in wacc["discord_ids"]: wacc["discord_ids"].append(discord_id)
+        if discord_id not in wacc["discord_ids"]:
+            wacc["discord_ids"].append(discord_id)
+            self.save()
         return wacc

main.py

@@ -63,7 +63,7 @@ class plugins:
 
 colors = framework.isobot.colors.Colors()
 currency_unused = framework.isobot.currency.CurrencyAPI(f'{wdir}/database/currency.json') # Initialize part of the framework (Currency)
-isobank = framework.isobank.manager.IsoBankManager(f"{wdir}/database/isobank/accounts.json", f"{wdir}/database/isobank/auth.json")
+# isobank = framework.isobank.manager.IsoBankManager(f"{wdir}/database/isobank/accounts.json", f"{wdir}/database/isobank/auth.json")
 isobankauth = framework.isobank.authorize.IsobankAuth(f"{wdir}/database/isobank/auth.json", f"{wdir}/database/isobank/accounts.json")
 
 #Events
@@ -993,6 +993,17 @@ async def repo(ctx:SlashContext):
     localembed = discord.Embed(title="Source-code Repositories", description="See and contribute to **isobot lazer's [GitHub repository](https://github.com/PyBotDevs/isobot-lazer)**\nSee our **[GitHub organization](https://github.com/PyBotDevs)**", color=discord.Color.random())
     await ctx.send(embed=localembed)
 
+@slash.slash(
+    name="isobank_register",
+    description="Registers a new IsoBank account with your Discord ID",
+    options=[
+        create_option(name="pin", description="Your new account's authentication ID. Must be a 6-digit integer.", option_type=4, required=True)
+    ]
+)
+async def isobank_register(ctx:SlashContext, pin:int):
+    isobankauth.register(ctx.author.id, pin)
+    await ctx.reply("Congratulations! Your new IsoBank account has been registered.", hidden=True)
+
 # Initialization
 utils.ping.host()
 client.run(api.auth.get_token())