Differential D5072 Diff 18261 swh/provenance/postgresql/provenance.py

Changeset View

Standalone View

swh/provenance/postgresql/provenance.py

	Show All 12 Lines
	from ..provenance import ProvenanceInterface			from ..provenance import ProvenanceInterface
	from ..revision import RevisionEntry			from ..revision import RevisionEntry
	from .db_utils import connect, execute_sql			from .db_utils import connect, execute_sql


	def normalize(path: bytes) -> bytes:			def normalize(path: bytes) -> bytes:
	return path[2:] if path.startswith(bytes("." + os.path.sep, "utf-8")) else path			return path[2:] if path.startswith(bytes("." + os.path.sep, "utf-8")) else path


	def create_database(conn: psycopg2.extensions.connection, conninfo: dict, name: str):
	conn.set_isolation_level(psycopg2.extensions.ISOLATION_LEVEL_AUTOCOMMIT)

	# Normalize dbname to avoid issues when reconnecting below
	name = name.casefold()

	# Create new database dropping previous one if exists
	cursor = conn.cursor()
	cursor.execute(f"""DROP DATABASE IF EXISTS {name}""")
	cursor.execute(f"""CREATE DATABASE {name}""")
	conn.close()

	# Reconnect to server selecting newly created database to add tables
	conninfo["dbname"] = name
	conn = connect(conninfo)

	sqldir = os.path.dirname(os.path.realpath(__file__))
	execute_sql(conn, os.path.join(sqldir, "provenance.sql"))


	########################################################################################			########################################################################################
	########################################################################################			########################################################################################
	########################################################################################			########################################################################################


	class ProvenancePostgreSQL(ProvenanceInterface):			class ProvenancePostgreSQL(ProvenanceInterface):
	def __init__(self, conn: psycopg2.extensions.connection):			def __init__(self, conn: psycopg2.extensions.connection):
	# TODO: consider adding a mutex for thread safety			# TODO: consider adding a mutex for thread safety
	▲ Show 20 Lines • Show All 456 Lines • Show Last 20 Lines