def setUp(self):
os.system(
"createdb peeweedbevolve_test && psql peeweedbevolve_test -c 'create extension IF NOT EXISTS hstore;' > /dev/null 2> /dev/null"
)
self.db = pwe.PostgresqlExtDatabase('peeweedbevolve_test')
self.db.connect()
peeweedbevolve.clear()
def get_handle(self):
if not self.handle or self.handle.is_closed():
db_host = self.get_env_var("CSW_HOST")
db_port = self.get_env_var("CSW_PORT")
db_user = self.get_env_var("CSW_USER")
db_password = self.get_env_var("CSW_PASSWORD")
self.handle = postgres_ext.PostgresqlExtDatabase(
"csw",
user=db_user,
password=db_password,
host=db_host,
port=db_port)
return self.handle
def execute_commands(self, commands, database="postgres"):
status = True
try:
db_host = self.get_env_var("CSW_HOST")
db_port = self.get_env_var("CSW_PORT")
db_user = self.get_env_var("CSW_USER")
db_password = self.get_env_var("CSW_PASSWORD")
db = postgres_ext.PostgresqlExtDatabase(database,
user=db_user,
password=db_password,
host=db_host,
port=db_port)
db.connect()
for command in commands:
cursor = db.execute_sql(command)
status &= cursor is not None
except Exception as err:
self.app.log.error("Failed to execute commands: " + str(err))
db.rollback()
status = False
db.close()
return status
"""Setup for the peewee database."""
from __future__ import annotations
import datetime
import enum
import typing
import peewee as pw
import playhouse.postgres_ext as pw_postgres
from . import config, utils
db = pw_postgres.PostgresqlExtDatabase(config.DB_NAME,
user=config.DB_USER,
password=config.DB_PASSWORD)
class BaseModel(pw.Model):
"""A base model, that sets the DB."""
class Meta:
"""Set the DB and use new table names."""
database = db
use_legacy_table_names = False
def __repr__(self, indent: int = 1) -> str: # pragma: no cover
"""Represent the model as a string."""
values = {}
for field in type(self)._meta.sorted_field_names:
values[field] = getattr(self, field)
import playhouse.postgres_ext as peewee
db = peewee.PostgresqlExtDatabase(database='postgres',
host='85.143.174.69',
port=15432,
user='******',
password='******',
autorollback=True,
register_hstore=True)
import playhouse.postgres_ext as peewee
db = peewee.PostgresqlExtDatabase(database='postgres',
host='localhost',
port=5432,
user='******',
password='******',
autorollback=True,
register_hstore=True)
import peewee
import playhouse.postgres_ext as pe
import yaml
import re
import os
import functools
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
CONFIG = os.path.join(BASE_DIR, '../config.yml')
with open(CONFIG, 'r') as yml:
cfg = yaml.load(yml.read(), Loader=yaml.FullLoader)
dbuser = cfg['dbuser']
dbpass = cfg['dbpass']
db = cfg['db']
psql_db = pe.PostgresqlExtDatabase(db, user=dbuser, password=dbpass)
def connect(func):
@functools.wraps(func)
def wrapper(*args, **kwargs):
try:
psql_db.connect()
return func(*args, **kwargs)
finally:
psql_db.close()
return wrapper
class Poll(peewee.Model):
title = peewee.TextField()
opened = peewee.DateTimeField()
close = peewee.DateTimeField()