karideo/back/src/db.py

from realog import debug

import psycopg2
import config

connection = None
connection_count = 0
def connect_bdd():
	global connection
	global connection_count
	if connection == None:
		debug.info("connect BDD: ")
		conf = config.get_rest_config()
		connection = psycopg2.connect(dbname=conf["db_name"], user=conf["db_user"], password=conf["db_password"], host=conf["db_host"], port=conf["db_port"])
	connection_count += 1
	return connection

def remove_connection():
	global connection
	global connection_count
	connection_count -= 1
	if connection_count < 0:
		debug.warning("Request remove too much time the BDD connection");
		connection_count = 0;
		return;
	if connection_count == 0:
		debug.warning("dicconnect BDD");
		connection.commit()
		connection.close()
		connection = None
		return;
	

base_bdd_name = "karideo_"