extensions.py

#!/usr/bin/env python
import requests
import json

from time import strftime
from datetime import datetime
from custom_httpauth import HTTPTokenAuth
from flask_sqlalchemy import SQLAlchemy
from flask_alembic import Alembic
from flask import jsonify, make_response, abort


db = SQLAlchemy()
auth = HTTPTokenAuth()
alembic = Alembic()
dateFormat = "%Y-%m-%d"
timeFormat = "%H:%M:%S"
dateTimeFormat = dateFormat + "T" + timeFormat


def dataResultSuccess(data, msg="", spuriousParameters=[], count=1, code=200):
    return make_response((jsonify({"success": True, "msg": msg, "spuriousparameters": spuriousParameters, "data": data, "count": count}), code))


def resultSuccess(msg="", spuriousParameters=[], code=200):
    return make_response((jsonify({"success": True, "msg": msg, "spuriousparameters": spuriousParameters}), code))


def resultFailure(msg, code):
    abort(make_response(jsonify({"success": False, "msg": msg}), code))


def getCurrenciesFromAPI():
    # call the currency API
    response = requests.get('http://openexchangerates.org/api/latest.json?app_id=79cbcd6eacd64256945500440eabcff9')
    # remove the b'' and \n     in the string (after converting the response to a string)
    responseString = str(response.content).replace(r"\n", "")
    responseString = responseString.replace(r"    ", "")
    responseString = responseString.lstrip('b')
    responseString = responseString.strip('\'')
    # get only the rates object (since we don't care about much else in the response)
    responseJson = json.loads(responseString)['rates']
    return responseJson


def queryToJson(queryParam):
    newQueryParam = "{"
    queryParam = queryParam.replace(' ', '')  # Remove all whitespaces
    queryList = queryParam.split(',')  # split the string into queries
    for part in queryList:
        subParts = part.split(':')  # further split things into key, value pairs
        newQueryParam += '"' + subParts[0] + '":"' + subParts[1] + '",'  # reconstruct the key value pairs in the new parameter
    newQueryParam = newQueryParam[:-1]  # remove the last trailing comma
    newQueryParam += '}'  # finish by adding the closing bracket
    return json.loads(newQueryParam)


def dateTimeNow():
    return strftime(dateTimeFormat)


def timeNow():
    return strftime(timeFormat)


def dateNow():
    return strftime.strftime(dateFormat)


def stringToDateTime(input):
    return datetime.strptime(input, '%Y-%m-%d')