49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
import sqlite3
|
|
from sqlite3 import Error
|
|
|
|
def init_conn(path):
|
|
conn = None
|
|
try:
|
|
conn = sqlite3.connect(path)
|
|
print ("Connection established!")
|
|
except Error as e:
|
|
print (e)
|
|
print ("Connection failed!")
|
|
return conn
|
|
|
|
def init_tables(connection):
|
|
sql = "CREATE TABLE IF NOT EXISTS messages( id integer PRIMARY KEY, nickname text NOT NULL, message text NOT NULL);"
|
|
connection.execute(sql)
|
|
|
|
def prepareDb(name):
|
|
conn = init_conn(name)
|
|
init_tables(conn)
|
|
conn.close()
|
|
|
|
def getMessages(name):
|
|
connection = init_conn(name)
|
|
sql = "SELECT * FROM messages;"
|
|
cursor = connection.cursor()
|
|
cursor.execute(sql)
|
|
rows = cursor.fetchall()
|
|
connection.close()
|
|
|
|
return rows
|
|
|
|
def generateMessagesHTMLTable(rows):
|
|
messagesTable = "<table border='1'>"
|
|
for row in rows:
|
|
messagesTable += "<tr>"
|
|
for cell in row:
|
|
messagesTable += "<td>" + str(cell) + "</td>"
|
|
messagesTable += "</tr>"
|
|
messagesTable += "</table>"
|
|
return messagesTable
|
|
|
|
def sendMessage(db, nickname, message):
|
|
connection = init_conn(db)
|
|
sql = "INSERT INTO messages(`nickname`, `message`) VALUES('{}', '{}')".format(nickname, message)
|
|
cursor = connection.cursor()
|
|
cursor.execute(sql)
|
|
connection.commit()
|
|
connection.close()
|