ddbb.py

import sqlite3
from threading import Lock
import time
import os
from base64 import b64encode
import json

settings = "/etc/friendspot/settings.json"
if not os.path.isfile(settings):
    settings = "settings.json"

with open(settings) as f:
    settings = json.loads(f.read())
    client_id = settings.get('client_id')
    auth = client_id + ":" + settings.get('client_secret')
    auth = b64encode(auth.encode('utf-8')).decode('utf-8')
    redirect_uri = settings.get('redirect_uri')


db = "/etc/friendspot/friendspot.db"
if not os.path.isfile(db):
    db = "friendspot.db"
db = sqlite3.connect(db, check_same_thread=False)
ldb = Lock()

cursor = db.cursor()
cursor.execute("CREATE TABLE IF NOT EXISTS user (id integer PRIMARY KEY NOT NULL, username text NOT NULL, name text NOT NULL, url text NOT NULL, img text NOT NULL, session text NOT NULL, access text NOT NULL, refresh text NOT NULL, valid int NOT NULL);")
cursor.execute("CREATE TABLE IF NOT EXISTS friends (user integer NOT NULL, friend integer NOT NULL, FOREIGN KEY (user) REFERENCES user(id), FOREIGN KEY (friend) REFERENCES user(id));")
db.commit()


def query(sql, *param):
    with ldb:
        cursor = db.cursor()
        cursor.execute(sql, param)
        result = cursor.fetchall()
        db.commit()
        return result


def queryone(sql, *param):
    with ldb:
        cursor = db.cursor()
        cursor.execute(sql, param)
        result = cursor.fetchone()
        db.commit()
        return result


def insert(sql, *param):
    with ldb:
        cursor = db.cursor()
        cursor.execute(sql, param)
        id = cursor.lastrowid
        db.commit()
        return id