import os
from sqlalchemy import *
from config import db

curdir = os.path.dirname(__file__)
if curdir == '':
    curdir = '.'

curdir = os.path.realpath(curdir)

sql_db = create_engine(db)
metadata = BoundMetaData(sql_db)

distance = Table('neighbours_distance', metadata,
                 Column('user', String(50)),
                 Column('neighbour', String(50)),
                 Column('distance', Float()))

try:
    distance.create()
except exceptions.SQLError:
    # already exists
    pass

def query(value):
    return sql_db.execute(value).fetchall()