def to_train_route(self):
train_route = TrainRoute()
train_route.line_name = self.line_name
train_route.stations = self.stations.split(';')
return train_route
def from_train_route(self, train_route):
"""
:type train_route : TrainRoute
"""
self.line_name = train_route.line_name
self.stations = ';'.join(train_route.stations)
train_routes_session = PostgresAccessorBase(TrainRouteModel,
__config.get_database_url())
def reset():
train_routes_session.query(TrainRouteModel).delete()
train_routes_session.commit()
def get_train_route_by_line(line_name):
"""
:rtype :TrainRoute
:param line_name:
"""
raw_train_route = train_routes_session.query(TrainRouteModel).filter(
TrainRouteModel.line_name == line_name).first()
return raw_train_route.to_train_route() if raw_train_route is not None else None
__author__ = 'rizkivmaster'
from sqlalchemy import Column, Date, String, BigInteger
from common.database.ModelBase import ModelBase
from common.database.PostgreBase import PostgresAccessorBase
from config import general_config
from enum import AccountingType
class Income(ModelBase):
__tablename__ = 'records'
date = Column(Date)
accountingId = Column(String, primary_key=True)
accountingPost = Column(String)
accountingType = Column(String)
notes = Column(String)
amount = Column(BigInteger)
def __init__(self):
self.date = None
self.accountingPost = None
self.accountingId = None
self.accountingType = None
self.notes = None
self.amount = None
income_accessors = PostgresAccessorBase(Income, general_config.get_database_url())
__author__ = 'rizkivmaster'
from sqlalchemy import Column, Date, String, BigInteger
from common.database.ModelBase import ModelBase
from common.database.PostgreBase import PostgresAccessorBase
from config import general_config
from enum import AccountingType
class Income(ModelBase):
__tablename__ = 'records'
date = Column(Date)
accountingId = Column(String, primary_key=True)
accountingPost = Column(String)
accountingType = Column(String)
notes = Column(String)
amount = Column(BigInteger)
def __init__(self):
self.date = None
self.accountingPost = None
self.accountingId = None
self.accountingType = None
self.notes = None
self.amount = None
income_accessors = PostgresAccessorBase(Income,
general_config.get_database_url())
from accessors.entity.BusEstimation import BusEstimation
from common import general_scheduler
from common.databases.PostgreBase import PostgresAccessorBase
from common.logging import logger_factory
from config import general_config
__bus_estimate_session = PostgresAccessorBase(BusEstimation, general_config.get_database_url())
__estimator_cache = dict()
__logger = logger_factory.create_logger(__name__)
def __refresh():
all_prediction = __get_all_bus_estimates()
for prediction in all_prediction:
assert isinstance(prediction, BusEstimation)
estimation_source = prediction.source
estimation_destination = prediction.destination
estimation_eta = prediction.eta
__add_sample(
estimation_source,
estimation_destination,
estimation_eta
)
__logger.info('Updated')
__scheduler = general_scheduler.schedule(general_config.get_eta_refresh_period(), __refresh)
# database part
def __get_all_bus_estimates():
def to_train_route(self):
train_route = TrainRoute()
train_route.line_name = self.line_name
train_route.stations = self.stations.split(';')
return train_route
def from_train_route(self, train_route):
"""
:type train_route : TrainRoute
"""
self.line_name = train_route.line_name
self.stations = ';'.join(train_route.stations)
train_routes_session = PostgresAccessorBase(TrainRouteModel,
__config.get_database_url())
def reset():
train_routes_session.query(TrainRouteModel).delete()
train_routes_session.commit()
def get_train_route_by_line(line_name):
"""
:rtype :TrainRoute
:param line_name:
"""
raw_train_route = train_routes_session.query(TrainRouteModel).filter(
TrainRouteModel.line_name == line_name).first()
return raw_train_route.to_train_route(
class Item(ModelBase):
__tablename__ = "Item"
id = Column(Integer, primary_key=True, autoincrement=True)
name = Column(String)
date_added = Column(Date)
price = Column(Integer)
image_url = Column(Date)
def __init__(self):
self.name = None
self.date_added = None
self.image_url = None
self.id = None
item_accessor = PostgresAccessorBase(Item, database_url=general_config.get_database_url())
#
#
# def get_item(id):
# return item_accessor.query(Item).filter(Item.id == id).first()
#
#
# def upset_item(item):
# """
# :type item:Item
# :param item:
# :return:
# """
# item_accessor.add(item) if get_item(item.id) else None
# item_accessor.commit()
#
def to_busway_transfer(self):
busway_transfer = BuswayTransfer()
busway_transfer.from_station = self.from_busway_station
busway_transfer.to_station = self.to_busway_station
return busway_transfer
def from_busway_transfer(self, busway_transfer):
"""
:type busway_transfer: BuswayTransfer
"""
self.index = index(busway_transfer)
self.from_busway_station = busway_transfer.from_station
self.to_busway_station = busway_transfer.to_station
busway_transfer_session = PostgresAccessorBase(BuswayTransferModel, __config.get_database_url())
def reset():
busway_transfer_session.query(BuswayTransferModel).delete()
busway_transfer_session.commit()
def get_busway_transfer(busway_transfer):
"""
:type busway_transfer: BuswayTransfer
:param busway_transfer:
:return:
"""
raw_busway_transfer = busway_transfer_session.query(BuswayTransferModel).filter(
BuswayTransferModel.index == index(busway_transfer)).first()
