2022-12-08 22:21:53 +01:00
import logging
import sqlite3
2022-12-10 01:32:50 +01:00
max_uses : int = 400
2022-12-08 22:21:53 +01:00
logging . basicConfig ( level = logging . INFO )
def debug ( message ) :
logging . info ( message )
#create a database called "database.db" if the database does not exist, else connect to it
conn = sqlite3 . connect ( ' ../database/data.db ' )
c = conn . cursor ( )
2022-12-10 01:03:34 +01:00
connp = sqlite3 . connect ( ' ../database/premium.db ' )
2022-12-10 01:27:14 +01:00
cp = connp . cursor ( )
2022-12-08 22:21:53 +01:00
# Create table called "data" if it does not exist with the following columns: guild_id, channel_id, api_key, is_active, max_tokens, temperature, frequency_penalty, presence_penalty, uses_count_today, prompt_size
2022-12-10 01:03:34 +01:00
c . execute ( ''' CREATE TABLE IF NOT EXISTS data (guild_id text, channel_id text, api_key text, is_active boolean, max_tokens integer, temperature real, frequency_penalty real, presence_penalty real, uses_count_today integer, prompt_size integer, prompt_prefix text, tts boolean, pretend_to_be text, pretend_enabled boolean) ''' )
2022-12-15 18:59:21 +01:00
cp . execute ( ''' CREATE TABLE IF NOT EXISTS data (user_id text, guild_id text, premium boolean) ''' )
# create table called "channels" if it does not exist with the following columns: guild_id, channel1, channel2, channel3, channel4, channel5
cp . execute ( ''' CREATE TABLE IF NOT EXISTS channels (guild_id text, channel1 text, channel2 text, channel3 text, channel4 text, channel5 text) ''' )
cp . execute ( " DELETE FROM channels " )
connp . commit ( )