aljaz/nutshell
1
0
Fork 0
mirror of https://github.com/aljazceru/nutshell.git synced 2025-12-22 03:24:18 +01:00
Code Issues Packages Projects Releases Wiki Activity

remove psycopg2

Browse Source
This commit is contained in:
callebtc
2022-10-03 01:02:27 +02:00
parent 5044315b0b
commit 0e24579862
7 changed files with 63 additions and 108 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
cashu/core/db.py
View File

@@ -85,43 +85,44 @@ class Database(Compat):
self.db_location_is_url = "://" in self.db_location
if self.db_location_is_url:
database_uri = self.db_location
raise Exception("Remote databases not supported. Use SQLite.")
# database_uri = self.db_location
if database_uri.startswith("cockroachdb://"):
self.type = COCKROACH
else:
self.type = POSTGRES
# if database_uri.startswith("cockroachdb://"):
# self.type = COCKROACH
# else:
# self.type = POSTGRES
import psycopg2 # type: ignore
# import psycopg2 # type: ignore
def _parse_timestamp(value, _):
f = "%Y-%m-%d %H:%M:%S.%f"
if not "." in value:
f = "%Y-%m-%d %H:%M:%S"
return time.mktime(datetime.datetime.strptime(value, f).timetuple())
# def _parse_timestamp(value, _):
# f = "%Y-%m-%d %H:%M:%S.%f"
# if not "." in value:
# f = "%Y-%m-%d %H:%M:%S"
# return time.mktime(datetime.datetime.strptime(value, f).timetuple())
psycopg2.extensions.register_type(
psycopg2.extensions.new_type(
psycopg2.extensions.DECIMAL.values,
"DEC2FLOAT",
lambda value, curs: float(value) if value is not None else None,
)
)
psycopg2.extensions.register_type(
psycopg2.extensions.new_type(
(1082, 1083, 1266),
"DATE2INT",
lambda value, curs: time.mktime(value.timetuple())
if value is not None
else None,
)
)
# psycopg2.extensions.register_type(
# psycopg2.extensions.new_type(
# psycopg2.extensions.DECIMAL.values,
# "DEC2FLOAT",
# lambda value, curs: float(value) if value is not None else None,
# )
# )
# psycopg2.extensions.register_type(
# psycopg2.extensions.new_type(
# (1082, 1083, 1266),
# "DATE2INT",
# lambda value, curs: time.mktime(value.timetuple())
# if value is not None
# else None,
# )
# )
psycopg2.extensions.register_type(
psycopg2.extensions.new_type(
(1184, 1114), "TIMESTAMP2INT", _parse_timestamp
)
)
# psycopg2.extensions.register_type(
# psycopg2.extensions.new_type(
# (1184, 1114), "TIMESTAMP2INT", _parse_timestamp
# )
# )
else:
if not os.path.exists(self.db_location):
print(f"Creating database directory: {self.db_location}")