4d9214e378
REF: Adjusted the Twitch Module to be on a 5 minute cooldown DOC: Adjusted related code for the new XP system in db and config
58 lines
1.3 KiB
Python
58 lines
1.3 KiB
Python
import sqlite3
|
|
|
|
|
|
def initialize_db():
|
|
conn = sqlite3.connect('selena.db')
|
|
cursor = conn.cursor()
|
|
|
|
# Birthdays table
|
|
cursor.execute('''
|
|
CREATE TABLE IF NOT EXISTS birthdays (
|
|
user_id TEXT PRIMARY KEY,
|
|
birthday TEXT
|
|
)
|
|
''')
|
|
|
|
# Currency table
|
|
cursor.execute('''
|
|
CREATE TABLE IF NOT EXISTS currency (
|
|
user_id TEXT PRIMARY KEY,
|
|
balance INTEGER,
|
|
last_earned TIMESTAMP
|
|
)
|
|
''')
|
|
|
|
# Followed channels table
|
|
cursor.execute('''
|
|
CREATE TABLE IF NOT EXISTS followed_channels (
|
|
twitch_name TEXT PRIMARY KEY,
|
|
discord_channel_id INTEGER
|
|
)
|
|
''')
|
|
cursor.execute('''
|
|
CREATE TABLE IF NOT EXISTS followed_youtube_channels (
|
|
youtube_channel_id TEXT PRIMARY KEY,
|
|
discord_channel_id INTEGER
|
|
)
|
|
''')
|
|
cursor.execute('''
|
|
CREATE TABLE IF NOT EXISTS youtube_status (
|
|
youtube_channel_id TEXT PRIMARY KEY,
|
|
last_video_id TEXT
|
|
)
|
|
''')
|
|
cursor.execute('''
|
|
CREATE TABLE IF NOT EXISTS user_xp (
|
|
user_id INTEGER PRIMARY KEY,
|
|
xp INTEGER,
|
|
level INTEGER
|
|
)
|
|
''')
|
|
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
|
|
def get_connection():
|
|
return sqlite3.connect('selena.db')
|