mirror of
https://github.com/Sheldan/Sissi.git
synced 2026-01-23 12:05:00 +00:00
199ca18cac
adding migration script for legacy quotes changing docker-compose file to use different container names
55 lines
1.9 KiB
Python
55 lines
1.9 KiB
Python
import sqlite3
|
|
from sqlite3 import Error
|
|
import datetime
|
|
|
|
from dto import LegacyQuote, LegacyQuoteAttachment
|
|
|
|
|
|
def create_connection(file: str):
|
|
conn = None
|
|
try:
|
|
conn = sqlite3.connect(file)
|
|
except Error as e:
|
|
print(e)
|
|
return conn
|
|
|
|
|
|
def load_all_quotes_with_attachments(conn):
|
|
cur = conn.cursor()
|
|
cur.execute("SELECT q.id, q.chan_id, q.author_id, "
|
|
"q.adder_id, q.jump_url, q.'timestamp', q.content,"
|
|
"a.msg_id, a.filename, a.url, a.is_image "
|
|
"FROM quotes q left outer join attachments a ON q.jump_url like '%' || a.msg_id")
|
|
|
|
rows = cur.fetchall()
|
|
|
|
quotes = {}
|
|
for row in rows:
|
|
quote_id = row[0]
|
|
if quote_id not in quotes:
|
|
quotes[quote_id] = LegacyQuote()
|
|
current_quote = quotes[quote_id]
|
|
current_quote.id = quote_id
|
|
current_quote.channel_id = row[1]
|
|
current_quote.author_id = row[2]
|
|
current_quote.adder_id = row[3]
|
|
if len(row[5]) > 25:
|
|
current_quote.creation_time_stamp = datetime.datetime.strptime(row[5], '%Y-%m-%d %H:%M:%S.%f')
|
|
elif len(row[5]) != 19:
|
|
current_quote.creation_time_stamp = datetime.datetime.strptime(row[5], '%Y-%m-%d %H:%M:%S.%f')
|
|
else:
|
|
current_quote.creation_time_stamp = datetime.datetime.strptime(row[5], '%Y-%m-%d %H:%M:%S')
|
|
current_quote.content = row[6]
|
|
current_quote.message_id = row[4][row[4].rindex('/')+1:]
|
|
|
|
if row[7] is not None:
|
|
if current_quote.attachments is None:
|
|
current_quote.attachments = []
|
|
attachment = LegacyQuoteAttachment()
|
|
attachment.message_id = row[7]
|
|
attachment.file_name = row[8]
|
|
attachment.url = row[9]
|
|
attachment.is_image = row[10] == 1
|
|
current_quote.attachments.append(attachment)
|
|
return quotes
|