73 lines
1.8 KiB
Python
73 lines
1.8 KiB
Python
import argparse
|
|
import sqlite3
|
|
import random
|
|
import csv
|
|
|
|
ap = argparse.ArgumentParser()
|
|
ap.add_argument('output', help='the file to output the database to')
|
|
ap.add_argument('input', help='a csv file to read from. if a row has one cell, that is the name of the entry. if a row has two cells, the first is the name of the entry, and the second is either a unicode emoji or a discord emoji id for the entry.')
|
|
ap.add_argument('-s', '--shuffle', help='shuffle entries. by default, entries are in the order they appear in the csv.', action='store_true')
|
|
|
|
a = ap.parse_args()
|
|
|
|
entries = []
|
|
|
|
# csv rows are either "name" or "name, emoji"
|
|
|
|
with open(a.input, newline='') as file:
|
|
reader = csv.reader(file)
|
|
for row in reader:
|
|
if len(row) == 0:
|
|
continue
|
|
if len(row) <= 2:
|
|
entries.append(list(map(lambda x: x.strip(), row)))
|
|
else:
|
|
print('encountered a row with more than two cells')
|
|
exit(1)
|
|
|
|
if a.shuffle:
|
|
random.shuffle(entries)
|
|
|
|
dbcon = sqlite3.connect(a.output)
|
|
|
|
cur = dbcon.cursor()
|
|
|
|
cur.execute('''
|
|
CREATE TABLE entries(
|
|
name TEXT UNIQUE NOT NULL,
|
|
emoji TEXT,
|
|
status INT NOT NULL,
|
|
last_poll_number INT NOT NULL,
|
|
sort_number INT UNIQUE NOT NULL
|
|
)
|
|
''')
|
|
|
|
cur.execute('''
|
|
CREATE TABLE polls(
|
|
poll_number INT UNIQUE NOT NULL,
|
|
entry_1 TEXT NOT NULL,
|
|
entry_2 TEXT NOT NULL,
|
|
posted INT NOT NULL,
|
|
expires INT NOT NULL,
|
|
channel_id TEXT NOT NULL,
|
|
message_id TEXT NOT NULL,
|
|
voters_1 TEXT,
|
|
voters_2 TEXT
|
|
)
|
|
''')
|
|
|
|
cur.execute('''
|
|
CREATE TABLE users(
|
|
id TEXT UNIQUE NOT NULL,
|
|
global_display TEXT NOT NULL,
|
|
avatar_hash TEXT,
|
|
retrieved INT NOT NULL
|
|
)
|
|
''')
|
|
|
|
for i, row in enumerate(entries):
|
|
name = row[0]
|
|
emoji = row[1] if len(row) > 1 else None
|
|
cur.execute('INSERT INTO entries VALUES(?, ?, 0, 0, ?)', (name, emoji, i))
|
|
|
|
dbcon.commit()
|