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()