db.py

import os

from sqlalchemy import create_engine
from sqlalchemy import Table, Column, Integer, String, Float, Boolean

from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base


Base = declarative_base()
engine = create_engine('sqlite:///' + os.getcwd() + '/data/store.db', echo=False)


class Business(Base):
    __tablename__ = 'businesses'
    id = Column(Integer, primary_key=True)
    bid = Column(String, unique=True)

    stars = Column(Integer)
    review_count = Column(Integer)
    latitude = Column(Float)
    longitude = Column(Float)

    Not = Column(Integer)
    good = Column(Integer)
    food = Column(Integer)
    place = Column(Integer)
    like = Column(Integer)
    out = Column(Integer)
    great = Column(Integer)
    very = Column(Integer)
    really = Column(Integer)
    about = Column(Integer)
    go = Column(Integer)
    time = Column(Integer)
    back = Column(Integer)
    service = Column(Integer)
    dont = Column(Integer)

    social_network_feature = Column(Float)
    success = Column(Integer)

    def __repr__(self):
        return '<Business bid=%s>' % (self.bid, )


class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    uid = Column(String, unique=True)

    review_count = Column(Integer)
    num_fans = Column(Integer)
    years_elite = Column(Integer)

    funny = Column(Integer)
    useful = Column(Integer)
    cool = Column(Integer)

    def __repr__(self):
        return '<User uid=%s>' % (self.uid, )


class Tip(Base):
    __tablename__ = 'tips'
    id = Column(Integer, primary_key=True)
    uid = Column(String)
    bid = Column(String)
    likes = Column(Integer)

    def __repr__(self):
        return '<Tip bid=%s uid=%s>' % (self.bid, self.uid)

class userAndRespect(object):
    def __init__(self, count,id):
	self.count = count
	self.id = id
	return
    def __cmp__(self, other):
	return cmp(self.count,other.count)

Base.metadata.create_all(engine)
Session = sessionmaker(bind=engine)

session = Session()