IRCat/modules/sqlite_local.py

23 lines
No EOL
1.2 KiB
Python

# IRCat module for local SQLite database (default)
import sqlite3, os
__ircat_type__ = "sql.provider" # The type of module
__ircat_requires__ = ["data-path"] # The required config.yml entries.
class broker:
def __init__(self, data_path):
if not os.path.isfile(data_path):
print("Creating database file...")
open(data_path, "w").write("")
self.conn = sqlite3.connect(data_path)
db = self.conn.cursor()
db.execute("""CREATE table IF NOT EXISTS nickserv (user varchar(255), modes varchar(255), hash varchar(255), cloak varchar(255))""")
db.execute("""CREATE table IF NOT EXISTS groups (name varchar(255), owner varchar(255))""")
db.execute("""CREATE table IF NOT EXISTS chanserv (name varchar(255), modes varchar(255), params varchar(255), owner varchar(255), usermodes varchar(255), optimodes varchar(255))""")
def nickserv_gethash(self, nick, password:str):
db = self.conn.cursor()
password = password.encode("UTF-8")
db.execute("SELECT * FROM nickserv WHERE user=?;", [nick])
e = db.fetchall()
if e == []:
return ["Nickname doesn't exist."]
else:
return e