Немножко привел к общему виду

This commit is contained in:
jar3b
2016-01-12 23:15:56 +03:00
committed by Jack Stdin
parent 1875b50a4c
commit 9803b1d25a
3 changed files with 111 additions and 73 deletions

47
aore/dbutils/dbhandler.py Normal file
View File

@@ -0,0 +1,47 @@
# -*- coding: utf-8 -*-
from aore.dbutils.dbimpl import DBImpl
from aore.config import db as dbparams
from aore.dbutils.dbschemas import db_shemas, allowed_tables
from traceback import format_exc
import psycopg2
import logging
class DbHandler:
def __init__(self):
logging.basicConfig(format='%(asctime)s %(message)s')
self.db = DBImpl(psycopg2, dbparams)
def bulk_csv(self, chunck_size, table_name, csv_file_name):
sql_query = "COPY \"{}\" ({}) FROM '{}' DELIMITER '\t' NULL 'NULL'". \
format(table_name,
", ".join(
db_shemas[table_name].fields),
csv_file_name)
try:
cur = self.db.get_cursor()
cur.execute(sql_query)
self.db.transaction_commit()
except:
self.db.transaction_rollback()
logging.error("Error updating sql. Reason : {}".format(format_exc()))
logging.warning("Inserted {} queries FROM {}".format(chunck_size, csv_file_name))
def pre_create(self):
f = open("aore/templates/postgre/pre_create.sql")
create_db_syntax = f.read()
f.close()
try:
cur = self.db.get_cursor()
cur.execute(create_db_syntax)
self.db.transaction_commit()
except:
self.db.transaction_rollback()
raise "Error downloading. Reason : {}".format(format_exc())
def pre_update(self):
# TODO: update actions
pass