Esempio n. 1
0
import os
import time
import psycopg2

from subprocess import check_call
from osmnames import settings
from osmnames import logger

log = logger.setup(__name__)


def exec_sql_from_file(filename,
                       user=settings.get("DB_USER"),
                       database=settings.get("DB_NAME"),
                       cwd=""):
    log.info("start executing sql file {}".format(filename))
    check_call([
        "psql", "-v", "ON_ERROR_STOP=1", "--username={}".format(user),
        "--dbname={}".format(settings.get("DB_NAME")), "--file={}/{}".format(
            cwd, filename)
    ],
               stdout=open(os.devnull, 'w'))
    log.info("finished executing sql file {}".format(filename))


def exec_sql(sql,
             user=settings.get('DB_USER'),
             database=settings.get('DB_NAME')):
    connection = psycopg2.connect(user=user, dbname=database)
    connection.set_session(autocommit=True)
    cursor = connection.cursor()
Esempio n. 2
0
import os
import time
import psycopg2

from subprocess import check_call
from osmnames import settings
from osmnames import logger

log = logger.setup(__name__)


def exec_sql_from_file(filename, user=settings.get("DB_USER"), database=settings.get("DB_NAME"), cwd=""):
    log.info("start executing sql file {}".format(filename))
    check_call([
            "psql",
            "-v", "ON_ERROR_STOP=1",
            "--username={}".format(user),
            "--dbname={}".format(settings.get("DB_NAME")),
            "--file={}/{}".format(cwd, filename)
        ], stdout=open(os.devnull, 'w')
    )
    log.info("finished executing sql file {}".format(filename))


def exec_sql(sql, user=settings.get('DB_USER'), database=settings.get('DB_NAME')):
    connection = psycopg2.connect(user=user, dbname=database)
    connection.set_session(autocommit=True)
    cursor = connection.cursor()
    cursor.execute(sql)
    return cursor