SQLite version of SugonCredit #1
1 changed files with 6 additions and 11 deletions
|
@ -34,18 +34,13 @@ class SugonCredit(commands.Cog):
|
||||||
return word + 's'
|
return word + 's'
|
||||||
|
|
||||||
def new_guild_generation(self, guild_id):
|
def new_guild_generation(self, guild_id):
|
||||||
"""Adds a new table for a guild to the SQLite databse."""
|
"""Adds a new table for a guild to the SQLite database."""
|
||||||
con = sqlite3.connect(f'{self.data_path}')
|
con = sqlite3.connect(self.data_path)
|
||||||
cur = con.cursor()
|
cur = con.cursor()
|
||||||
exist_check = cur.execute(f'''IF EXISTS
|
try:
|
||||||
(SELECT object_id FROM sys.tables
|
cur.execute(f"SELECT 1 FROM {guild_id} LIMIT 1;")
|
||||||
WHERE name = '{guild_id}'
|
except sqlite3.OperationalError:
|
||||||
AND SCHEMA_NAME(schema_id) = 'dbo')
|
cur.execute(f"CREATE TABLE {guild_id} (user_id TEXT, balance REAL);")
|
||||||
PRINT 'False'
|
|
||||||
ELSE
|
|
||||||
PRINT 'True';''')
|
|
||||||
if exist_check == False:
|
|
||||||
cur.execute(f'''CREATE TABLE '{guild_id}' (user_id text, balance real)''')
|
|
||||||
con.commit()
|
con.commit()
|
||||||
con.close()
|
con.close()
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue