def mget_map_by_platform_and_time_after(cls, platform, time_after):
session = DBSession()
subqry = session.query(func.min(cls.time_at)).\
filter(cls.time_at >= time_after, cls.platform == platform)
result = session.query(cls). \
filter(cls.platform == platform, cls.time_at == subqry).all()
session.close()
return {_.tv_name: _.all_play_counts for _ in result}
def get_by_openid(cls, openid):
session = DBSession()
result = session.query(cls).filter(cls.openid == openid).first()
if not result:
raise(ttypes.UserException(
code=1,
message='空'
))
session.close()
return result
def login(cls, username, passwd):
session = DBSession()
result = session.query(cls).filter(cls.username == username and
cls.password == passwd).first()
if not result:
raise(ttypes.UserException(
code=1,
message='空'
))
session.close()
return result
def mget_map_by_platform_and_time_after(cls, platform, time_after):
session = DBSession()
subqry = session.query(func.min(cls.time_at)).\
filter(cls.time_at >= time_after, cls.platform == platform)
result = session.query(cls). \
filter(cls.platform == platform, cls.time_at == subqry).all()
session.close()
return {_.tv_name: _.all_play_counts for _ in result}
def add_all(cls, articles):
if not articles:
return
session = DBSession()
session.add_all(articles)
try:
session.flush()
session.commit()
except SQLAlchemyError:
session.rollback()
raise(SQLAlchemyError)
finally:
session.close()
def mget_by_parent_ids(cls, parent_ids=[]):
session = DBSession()
result = session.query(cls).filter(cls.parent_id.in_(parent_ids)).all()
session.close()
return result
def mget_by_platform_and_type(cls, platform, type):
session = DBSession()
result = session.query(cls).\
filter(cls.platform == platform and cls.type == type).all()
session.close()
return result
def mget(cls):
session = DBSession()
result = session.query(cls).all()
session.close()
return result
def update(cls, **kwds):
session = DBSession()
session.query(cls).filter(
cls.name == kwds['name'], cls.platform == kwds['platform'],
cls.type == kwds['type']).update(kwds) # noqa
try:
session.flush()
session.commit()
except SQLAlchemyError:
session.rollback()
raise (SQLAlchemyError)
finally:
session.close()
def mget_by_platform(cls, platform):
session = DBSession()
result = session.query(cls).filter(cls.platform == platform).all()
session.close()
return result
def mget_by_platform_and_type(cls, platform, type):
session = DBSession()
result = session.query(cls).\
filter(cls.platform == platform and cls.type == type).all()
session.close()
return result
def update_all(cls, moods):
session = DBSession()
for mood in moods:
session.query(cls). \
filter(cls.time_at == mood.time_at). \
filter(cls.city_id == mood.city_id). \
update({
cls.happy: mood.happy,
cls.good: mood.good,
cls.anger: mood.anger,
cls.sorrow: mood.sorrow,
cls.fear: mood.fear,
cls.evil: mood.evil,
cls.surprise: mood.surprise,
cls.cps: mood.cps,
})
try:
session.flush()
session.commit()
except SQLAlchemyError:
session.rollback()
raise(SQLAlchemyError)
finally:
session.close()
def update(cls, **kwds):
session = DBSession()
session.query(cls).filter(cls.tv_id == kwds['tv_id']). \
update(kwds)
try:
session.flush()
session.commit()
except SQLAlchemyError:
session.rollback()
raise(SQLAlchemyError)
finally:
session.close()
def update(cls, **kwds):
session = DBSession()
session.query(cls).filter(cls.name == kwds['name'],
cls.platform == kwds['platform'], cls.type == kwds['type']).update(kwds) # noqa
try:
session.flush()
session.commit()
except SQLAlchemyError:
session.rollback()
raise(SQLAlchemyError)
finally:
session.close()
def mget(cls, offset=0, limit=10):
session = DBSession()
result = session.query(cls).all()
session.close()
return result
def mget(cls, query):
session = DBSession()
result = session.query(cls).order_by(cls.time.desc()). \
offset(query.offset).limit(query.limit).all()
session.close()
return result
def register(cls, **kwds):
session = DBSession()
user = cls(**kwds)
session.add(user)
user_id = None
try:
session.flush()
session.commit()
user_id = user.id
except SQLAlchemyError:
session.rollback()
raise(ttypes.UserException)(
code=2,
message='数据增加失败'
)
finally:
session.close()
return user_id
def add_all(cls, hot_words):
if not hot_words:
return
session = DBSession()
session.add_all(hot_words)
try:
session.flush()
session.commit()
except SQLAlchemyError:
session.rollback()
# raise(SQLAlchemyError)
finally:
session.close()
def update(cls, city_id, time_at, word, times):
session = DBSession()
session.query(cls). \
filter(cls.time_at == time_at). \
filter(cls.city_id == city_id). \
filter(cls.word == word). \
update({
cls.times: times,
})
try:
session.flush()
session.commit()
except SQLAlchemyError:
session.rollback()
raise(SQLAlchemyError)
finally:
session.close()
def add(cls, **kwds):
session = DBSession()
session.add(cls(**kwds))
try:
session.flush()
session.commit()
except SQLAlchemyError:
session.rollback()
raise(SQLAlchemyError)
finally:
session.close()
def update(cls, **kwds):
session = DBSession()
session.query(cls).filter(cls.tv_id == kwds['tv_id']). \
update(kwds)
try:
session.flush()
session.commit()
except SQLAlchemyError:
session.rollback()
raise (SQLAlchemyError)
finally:
session.close()
def mget_by_platform(cls, platform):
session = DBSession()
result = session.query(cls).filter(cls.platform == platform).all()
session.close()
return result
def add(cls, **kwds):
session = DBSession()
session.add(cls(**kwds))
try:
session.flush()
session.commit()
except SQLAlchemyError:
session.rollback()
raise (SQLAlchemyError)
finally:
session.close()
def mget(cls):
session = DBSession()
result = session.query(cls).all()
session.close()
return result
def add_all(cls, moods):
session = DBSession()
session.add_all(moods)
try:
session.flush()
session.commit()
except SQLAlchemyError:
session.rollback()
raise(SQLAlchemyError)
finally:
session.close()
