py.lib

     1.1 --- a/db/sqlite.py	Sat Aug 13 20:06:47 2022 +0300
     1.2 +++ b/db/sqlite.py	Sun Aug 14 12:19:08 2022 +0300
     1.3 @@ -2,35 +2,36 @@
     1.4  import sqlite3
     1.5  from sqlite3 import Error, IntegrityError
     1.6  
     1.7 +
     1.8  class DB:
     1.9 -	def __init__(self, db_file):
    1.10 -		self._conn = sqlite3.connect(db_file)
    1.11 -		self._ex = self._conn.execute
    1.12 -		self.commit = self._conn.commit
    1.13 -		self.rollback = self._conn.rollback
    1.14 -		
    1.15 -		# DB PREP
    1.16 -		self._ex("PRAGMA journal=WAL")
    1.17 -		self._ex("PRAGMA foreign_keys=ON")
    1.18 -		self.commit()
    1.19 -	
    1.20 -	def __del__(self):
    1.21 -		try:
    1.22 -			self.rollback()     
    1.23 -			self._conn.close()
    1.24 -		except:
    1.25 -			pass
    1.26 -	
    1.27 -	def __call__(self, *a, **kwa):
    1.28 -		cur = self._conn.cursor()
    1.29 -		cur.execute(*a, **kwa)
    1.30 -		return cur
    1.31 -	
    1.32 -	def cq(self, *a, **wa):
    1.33 -		try:
    1.34 -			res = self(*a, **wa)
    1.35 -			self.commit()
    1.36 -			return res
    1.37 -		except Error as e:
    1.38 -			self.rollback()
    1.39 -			raise e
    1.40 +    def __init__(self, db_file):
    1.41 +        self._conn = sqlite3.connect(db_file)
    1.42 +        self._ex = self._conn.execute
    1.43 +        self.commit = self._conn.commit
    1.44 +        self.rollback = self._conn.rollback
    1.45 +
    1.46 +        # DB PREP
    1.47 +        self._ex("PRAGMA journal=WAL")
    1.48 +        self._ex("PRAGMA foreign_keys=ON")
    1.49 +        self.commit()
    1.50 +
    1.51 +    def __del__(self):
    1.52 +        try:
    1.53 +            self.rollback()
    1.54 +            self._conn.close()
    1.55 +        except:
    1.56 +            pass
    1.57 +
    1.58 +    def __call__(self, *a, **kwa):
    1.59 +        cur = self._conn.cursor()
    1.60 +        cur.execute(*a, **kwa)
    1.61 +        return cur
    1.62 +
    1.63 +    def cq(self, *a, **wa):
    1.64 +        try:
    1.65 +            res = self(*a, **wa)
    1.66 +            self.commit()
    1.67 +            return res
    1.68 +        except Error as e:
    1.69 +            self.rollback()
    1.70 +            raise e