from database import DbConnection
CONFIG_FILE = "ingress.conf"
config = configparser.ConfigParser()
config.read(CONFIG_FILE)
LOGFILE = config['logging']['file']
LISTEN = config['carbon']['listen']
PORT = config['carbon'].getint('port')
logging.basicConfig(filename=LOGFILE,
level=logging.DEBUG,
format="%(asctime)s %(levelname)s %(message)s")
db = DbConnection()
class RequestHandler(socketserver.StreamRequestHandler):
def handle(self):
line = str(self.rfile.readline().strip(), "utf8")
logging.debug(line)
fields = line.split()
logging.debug(fields)
sensorid, channel = fields[0].split(".")
value = float(fields[1])
timestamp = int(fields[2])
dp = DataPoint(sensorid=sensorid,
channel=channel,
value=value,
timestamp=datetime.datetime.fromtimestamp(
def test_database_connection(db_credentials):
credentials = db_credentials
cursor = DbConnection(**credentials).connect_to_db().cursor();
cursor.execute("SELECT current_user")
assert cursor.fetchone()[0] == 'docker'
import json
import os
import uuid
app = Flask(__name__)
UPLOAD_PATH = 'UPLOAD_PATH'
ALLOWED_EXTENSIONS = 'ALLOWED_EXTENSIONS'
DIRNAME = os.path.dirname(os.path.abspath(__file__))
app.config[UPLOAD_PATH] = 'static'
app.config[ALLOWED_EXTENSIONS] = {'png', 'jpg', 'jpeg'}
with open('dbconfig.json') as config_file:
db_config = json.load(config_file)
db = DbConnection(db_config, app)
tokens = []
# TODO(remove indents in production)
def to_json(obj):
return json.dumps(obj,
default=lambda o: o.__dict__,
ensure_ascii=False,
indent=4).encode('utf-8')
def login_required(f):
@wraps(f)
def wrap(*args, **kwargs):
token = request.headers.get('token')
def connection(db_credentials): return DbConnection(**db_credentials).connect_to_db()