| Line | |
|---|
| 1 | import os |
|---|
| 2 | from sqlalchemy import * |
|---|
| 3 | from config import db |
|---|
| 4 | |
|---|
| 5 | curdir = os.path.dirname(__file__) |
|---|
| 6 | if curdir == '': |
|---|
| 7 | curdir = '.' |
|---|
| 8 | |
|---|
| 9 | curdir = os.path.realpath(curdir) |
|---|
| 10 | |
|---|
| 11 | sql_db = create_engine(db) |
|---|
| 12 | metadata = BoundMetaData(sql_db) |
|---|
| 13 | |
|---|
| 14 | distance = Table('neighbours_distance', metadata, |
|---|
| 15 | Column('user', String(50)), |
|---|
| 16 | Column('neighbour', String(50)), |
|---|
| 17 | Column('distance', Float())) |
|---|
| 18 | |
|---|
| 19 | try: |
|---|
| 20 | distance.create() |
|---|
| 21 | except exceptions.SQLError: |
|---|
| 22 | # already exists |
|---|
| 23 | pass |
|---|
| 24 | |
|---|
| 25 | def query(value): |
|---|
| 26 | return sql_db.execute(value).fetchall() |
|---|
| 27 | |
|---|
