def psycopg2ct_uri_no_port_connect():
conn_uri_no_port = psycopg2ct.connect(
"postgresql://%s:%s@%s/%s" % (PG_CONF['user'], PG_CONF['password'],
PG_CONF['host'], PG_CONF['database']))
sql = common_select_sql()
res = common_select_conn(conn_uri_no_port, sql)
return res
def psycopg2ct_connectstring_connect():
conn_connectstring = psycopg2ct.connect(
"dbname=%s user=%s password=%s host=%s port=%s" %
(PG_CONF['database'], PG_CONF['user'], PG_CONF['password'],
PG_CONF['host'], PG_CONF['port']))
sql = common_select_sql()
res = common_select_conn(conn_connectstring, sql)
return res
def psycopg2ct_no_support_connect_two():
conn_no_support_two = psycopg2ct.connect(
"postgresql://%s:%s@[%s]:%s/%s" %
(PG_CONF['user'], PG_CONF['password'], PG_CONF['IPv6_address'],
PG_CONF['port'], PG_CONF['database']))
sql = common_select_sql()
res = common_select(conn_no_support_two, sql)
return res
def psycopg2ct_no_support_connect():
conn_no_support = psycopg2ct.connect(
"postgresql:///%s?host=%s&port=%s&user=%s&password=%s" %
(PG_CONF['database'], PG_CONF['host'], PG_CONF['port'],
PG_CONF['user'], PG_CONF['password']))
sql = common_select_sql()
res = common_select(conn_no_support, sql)
return res
def psycopg2ct_no_port_connect():
conn_no_port = psycopg2ct.connect(host=PG_CONF['host'],
user=PG_CONF['user'],
password=PG_CONF['password'],
database=PG_CONF['database'])
sql = common_select_sql()
res = common_select_conn(conn_no_port, sql)
return res
def test_datetime_tz_roundtrip(self):
import datetime
import psycopg2ct
conn = psycopg2ct.connect(self.dsn)
cur = conn.cursor()
tz = psycopg2ct.tz.FixedOffsetTimezone(8 * 60)
d = (datetime.datetime(2010, 05, 03, 10, 20, 30, tzinfo=tz),)
def save_to_db():
try:
rConn = pg.connect(database="hotstats",
user="******",
password="******",
host="YOUR_SERVER",
port="5432")
cursor = rConn.cursor()
rConn.autocommit = True
except Exception, e:
print "Error while trying to establish connection with database %s" % e
def test_connect(self):
import psycopg2ct
conn = psycopg2ct.connect(
database=self.database, user=self.user, password=self.password)
conn.close()
conn = psycopg2ct.connect(
database=self.database, user=self.user, password=self.password,
port=5432)
conn.close()
conn = psycopg2ct.connect(
database=self.database, user=self.user, password=self.password,
port="5432")
conn.close()
with self.assertRaises(TypeError):
psycopg2ct.connect(database=self.database, user=self.user,
password=self.password, port=object())
def save_data(replayData, path):
# read credential file
with open('credentials.json') as data_file:
conf = json.load(data_file)
try:
rConn = psycopg2ct.connect(database=conf['database'],
user=conf['user'],
password=conf['password'],
host=conf['host'],
port=conf['port'])
cursor = rConn.cursor()
rConn.autocommit = True
except Exception, e:
print "Error while trying to establish connection with database %s" % e
def _start_new_connection():
global conn
global conn_pool
global conn_pool
conn = psycopg2ct.connect(host=PG_CONF['host'],
port=PG_CONF['port'],
user=PG_CONF['user'],
password=PG_CONF['password'],
database=PG_CONF['database'])
conn_pool = psycopg2ct.pool.SimpleConnectionPool(
minconn=PG_CONF['minconn'],
maxconn=PG_CONF['maxconn'],
host=PG_CONF['host'],
port=PG_CONF['port'],
user=PG_CONF['user'],
password=PG_CONF['password'],
database=PG_CONF['database'])
conn_pool = conn_pool.getconn()
def connect(self):
import psycopg2ct
return psycopg2ct.connect(self.dsn)
# -*- coding: utf-8 -*-
import psycopg2ct
import psycopg2ct.pool
from public.db_util import (common_insert_sql, common_update_sql,
common_select_sql, common_delete_sql,
common_table_error_sql, common_select,
common_execute, psycopg2_timeout_sql,
psycopg2_sleep_sql, common_field_error_sql,
common_syntax_error_sql, common_select_conn)
from ..config import PG_CONF
conn = psycopg2ct.connect(host=PG_CONF['host'],
port=PG_CONF['port'],
user=PG_CONF['user'],
password=PG_CONF['password'],
database=PG_CONF['database'])
conn_pool = psycopg2ct.pool.SimpleConnectionPool(minconn=PG_CONF['minconn'],
maxconn=PG_CONF['maxconn'],
host=PG_CONF['host'],
port=PG_CONF['port'],
user=PG_CONF['user'],
password=PG_CONF['password'],
database=PG_CONF['database'])
conn_pool = conn_pool.getconn()
def psycopg2ct_no_support_connect():
