-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsettings.py
More file actions
66 lines (50 loc) · 2.05 KB
/
settings.py
File metadata and controls
66 lines (50 loc) · 2.05 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
import aiosqlite
async def database():
conn = await aiosqlite.connect('settings.db')
c = await conn.cursor()
await c.execute('''CREATE TABLE IF NOT EXISTS settings
(guild_id TEXT, scam_delete TEXT DEFAULT 'false', verified_role TEXT, counting_channel TEXT)''')
await conn.commit()
await conn.close()
async def set_scam_delete(guild_id, scam_delete):
conn = await aiosqlite.connect('settings.db')
c = await conn.cursor()
await c.execute("SELECT * FROM settings WHERE guild_id = ?", (guild_id,))
result = await c.fetchone()
if result is None:
await c.execute("INSERT INTO settings (guild_id, scam_delete) VALUES (?, ?)", (guild_id, scam_delete))
else:
await c.execute("UPDATE settings SET scam_delete = ? WHERE guild_id = ?", (scam_delete, guild_id))
await conn.commit()
await conn.close()
async def get_scam_delete(guild_id):
conn = await aiosqlite.connect('settings.db')
c = await conn.cursor()
await c.execute("SELECT scam_delete FROM settings WHERE guild_id = ?", (guild_id,))
result = await c.fetchone()
await conn.close()
if result is not None:
scam_delete = result[0]
return scam_delete
return None
async def set_verified_role(guild_id, role_id):
conn = await aiosqlite.connect('settings.db')
c = await conn.cursor()
await c.execute("SELECT * FROM settings WHERE guild_id = ?", (guild_id,))
result = await c.fetchone()
if result is None:
await c.execute("INSERT INTO settings (guild_id, verified_role) VALUES (?, ?)", (guild_id, role_id))
else:
await c.execute("UPDATE settings SET verified_role = ? WHERE guild_id = ?", (role_id, guild_id))
await conn.commit()
await conn.close()
async def get_verified_role(guild_id):
conn = await aiosqlite.connect('settings.db')
c = await conn.cursor()
await c.execute("SELECT verified_role FROM settings WHERE guild_id = ?", (guild_id,))
result = await c.fetchone()
await conn.close()
if result is not None:
role_id = result[0]
return role_id
return None