29 lines
No EOL
1.5 KiB
Python
29 lines
No EOL
1.5 KiB
Python
# IRCat module for local SQLite database (default)
|
|
import sqlite3, os, traceback
|
|
from cryptography.fernet import Fernet
|
|
__ircat_type__ = "sql.provider" # The type of module
|
|
__ircat_requires__ = ["data-path", "fernet-key"] # The required config.yml entries.
|
|
class broker:
|
|
def __init__(self, data_path, fernet_key):
|
|
if not os.path.isfile(data_path):
|
|
print("Creating database file...")
|
|
open(data_path, "w").write("")
|
|
self.conn = sqlite3.connect(data_path, check_same_thread=False)
|
|
self.fnet = Fernet(fernet_key)
|
|
db = self.conn.cursor()
|
|
db.execute("""CREATE table IF NOT EXISTS bans (ip varchar(255), reason varchar(255))""")
|
|
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_identify(self, nick, password:str):
|
|
db = self.conn.cursor()
|
|
db.execute("SELECT * FROM nickserv WHERE user=?;", [nick])
|
|
e = db.fetchall()
|
|
if e == []:
|
|
return False
|
|
else:
|
|
try:
|
|
return e[0] if self.fnet.decrypt(e[0][2]) == password else False
|
|
except:
|
|
print(traceback.format_exc())
|
|
return False |