def config_fixture() -> Config:
return Config(
data_refresh_period=None,
reverse_geocoding_path=TEST_DATA_PATH / "station_location.csv",
port=0,
user_content_storage=None,
velib_data_base_path=TEST_DATA_PATH,
)
def uri(self, type):
db_uri = None
if type == DBType.sqlite:
db_uri = "sqlite:///" + Config()["database"]["sqlite"]["path"]
if type == DBType.mysql:
config = Config()["database"]["mysql"]
db_uri = "mysql+pymysql://{0}:{1}@{2}:3306/{3}".format(
config["user"], config["password"], config["host"],
config["db"])
if type == DBType.postgres:
config = Config()["database"]["postgres"]
db_uri = "postgresql://{0}:{1}@{2}:{3}/{4}".format(
config["user"], config["password"], config["host"],
config["port"], config["db"])
return db_uri
def test_init_results(self):
"""Make sure _init_results does what is expected"""
config = Config()
config.monitors = {"http://test/": {}, "http://test2/": {}}
result = App._init_results(config)
assert result["http://test/"] is None
assert result["http://test2/"] is None
def test_expected_configs(self):
expected_region = 'test-region'
expected_sender_email = '*****@*****.**'
os.environ['SES_AWS_REGION'] = expected_region
os.environ['SES_SENDER_EMAIL'] = expected_sender_email
expected = Config(
ses_aws_region=expected_region,
ses_sender_email=expected_sender_email,
)
actual = get_configs_by_env()
self.assertEqual(expected, actual)
def api_config_post():
args = request.form
name = args["name"]
api = args["api"]
token = args["token"]
config = Config.query.get(1)
if not config:
config = Config(name, api, token)
db.session.add(config)
db.session.commit()
config.name = name
config.api = api
config.token = token
db.session.commit()
return jsonify({"message": "OK."}), 200
#!/usr/bin/env python3
import logging
from app import Config
from ldaploginprovider import LDAPLoginProvider
LOGFORMAT = "[%(asctime)s] [%(levelname)8s] --- %(message)s (%(filename)s:%(lineno)s)"
logging.basicConfig(level=logging.INFO, format=LOGFORMAT)
config = Config()
logging.info("Building ldap_provider")
ldap_provider = LDAPLoginProvider(config.ldap_server, config.ldap_base_dn, config.ldap_user_prefix)
ldap_provider.build_connection()
logging.info("Attempting to login")
user = ldap_provider.login(config.known_user, config.known_pass)
logging.info("Login: " + str(user.__dict__))
def api_config_get():
config = Config.query.get(1)
if not config:
config = Config("", "", "")
db.session.commit()
return jsonify({"config": config.to_dict()}), 200
def get_service(self, service):
return self.c.catalog.service(service)
def deregister_service(self, service_id):
return self.c.agent.service.deregister(service_id)
def register_service(self, service_name, vmid, port, address, tags):
# register(name, service_id=None, address=None, port=None, tags=None, check=None, token=None, script=None, interval=None, ttl=None, http=None, timeout=None, enable_tag_override=False)
self.c.agent.service.register(service_name,
service_id=vmid,
port=port,
address=address,
tags=[tags])
config = Config().read()
logging.basicConfig(filename=config['settings']['log_file'],
filemode='a',
format='%(asctime)s [ %(levelname)s ] - %(message)s',
datefmt='%m/%d/%Y %H:%M:%S',
level=config['settings']['log_level'])
def register(service_name, data):
new = []
for x in data:
new.append(x['vmid'])
old = []
for x in Consul().get_service(service_name)[1]:
old.append(x['ServiceID'])
from sqlalchemy.exc import DatabaseError
from flask_sqlalchemy import Model
from sqlalchemy.orm import sessionmaker
from app import Config
from app.utils.traces import print_exception_traces
class DBType(Enum):
sqlite = 1
mysql = 2
postgres = 3
db_type = DBType.sqlite # Default
if "sqlite" == Config()["database"]["active"]:
db_type = DBType.sqlite
if "mysql" == Config()["database"]["active"]:
db_type = DBType.mysql
if "postgres" == Config()["database"]["active"]:
db_type = DBType.postgres
## Init session
class DBSession():
def uri(self, type):
db_uri = None
if type == DBType.sqlite:
db_uri = "sqlite:///" + Config()["database"]["sqlite"]["path"]
def interface(ctx):
"""SetupEnv: A simple templating program"""
ctx.obj = Config()
def test_read_example(self):
"""Just try to read the example config and make sure we do it right"""
config = Config()
config.read_from_file('config.example.yaml')
def setUpClass(cls):
# create test database
config = Config(RuntimeEnvironment.TEST)
if not database_exists(config.db_uri):
create_database(config.db_uri)
def print_exception_traces(e):
config = Config()
if config["stacktrace"]:
print(traceback.format_exc())
else:
print(e)
