class CreditModel(banco.Model):
__tablename__ = 'credit'
credit_id = banco.Column(banco.Integer, primary_key=True)
value = banco.Column(banco.Float, nullable=False)
description = banco.Column(banco.String, nullable=False)
account_id = banco.Column(banco.Integer,
banco.ForeignKey('account.account_id'))
def __init__(self, account_id, value, description):
self.account_id = account_id
self.value = value
self.description = description
def json(self):
return {
"credit_id": self.credit_id,
"account_number": self.account_id,
"value": self.value,
"description": self.description,
}
@classmethod
def find_credits(cls, account_id, limit=50, offset=0):
return cls.query.filter_by(
account_id=account_id).limit(limit).offset(offset)
@classmethod
def find_all_credits(cls, account_id):
return cls.query.filter_by(account_id=account_id)
def save(self):
banco.session.add(self)
banco.session.commit()
class HotelModel(banco.Model):
__tablename__ = 'hoteis'
hotel_id = banco.Column(banco.String(30), primary_key=True)
nome = banco.Column(banco.String(80))
estrelas = banco.Column(banco.Float(precision=1))
diaria = banco.Column(banco.Float(precision=2))
cidade = banco.Column(banco.String(40))
site_id = banco.Column(banco.Integer, banco.ForeignKey('sites.site_id'))
#site = banco.relashionship('SiteModel')
def __init__(self, hotel_id, nome, estrelas, diaria, cidade, site_id):
self.hotel_id = hotel_id
self.nome = nome
self.estrelas = estrelas
self.diaria = diaria
self.cidade = cidade
self.site_id = site_id
#S
#converte em json
def json(self):
return {
'hotel_id': self.hotel_id,
'nome': self.nome,
'estrelas': self.estrelas,
'diaria': self.diaria,
'cidade': self.cidade,
'site_id': self.site_id
}
@classmethod
def find_hotel(cls, hotel_id):
hotel = cls.query.filter_by(
hotel_id=hotel_id).first() #select * from w
if hotel:
return hotel
return None
@classmethod
def find_hoteis(cls):
hoteis = cls.query.all() #select * from
return hoteis
def save_hotel(self):
banco.session.add(self)
banco.session.commit()
def update(self, nome, estrelas, diaria, cidade):
self.nome = nome
self.estrelas = estrelas
self.diaria = diaria
self.cidade = cidade
def delete(self):
banco.session.delete(self)
banco.session.commit()
class HotelModel(banco.Model):
__tablename__ = 'hotels'
id = banco.Column(banco.Integer, primary_key=True, autoincrement=True)
name = banco.Column(banco.String(80))
stars = banco.Column(banco.Float(precision=1))
daily = banco.Column(banco.Float(precision=2))
state = banco.Column(banco.String(80))
city = banco.Column(banco.String(80))
site_id = banco.Column(banco.Integer, banco.ForeignKey('sites.id'))
def __init__(self, name, stars, daily, state, city, site_id):
self.name = name
self.stars = stars
self.daily = daily
self.state = state
self.city = city
self.site_id = site_id
def json(self):
return {
'id': self.id,
'name': self.name,
'stars': self.stars,
'daily': self.daily,
'state': self.state,
'city': self.city,
'site_id': self.site_id
}
@classmethod
def hotel_find_by_name(cls, name):
hotel = cls.query.filter_by(name=name).first()
if hotel:
return hotel
return None
@classmethod
def hotel_find_by_id(cls, id):
hotel = cls.query.filter_by(id=id).first()
if hotel:
return hotel
return None
def save_hotel(self):
banco.session.add(self)
banco.session.commit()
def update_hotel(self, name, stars, daily, state, city, site_id):
self.name = name
self.stars = stars
self.daily = daily
self.state = state
self.city = city
self.site_id = site_id
def delete_hotel(self):
banco.session.delete(self)
banco.session.commit()
class HotelModel(banco.Model):
__tablename__ = 'hoteis'
hotel_pk = banco.Column(banco.Integer, primary_key=True)
hotel_id = banco.Column(banco.String)
nome = banco.Column(banco.String(80))
estrelas = banco.Column(banco.Float(precision=1))
diaria = banco.Column(banco.Float(precision=2))
cidade = banco.Column(banco.String(40))
site_id = banco.Column(banco.Integer, banco.ForeignKey('sites.site_id'))
def __init__(self, hotel_id, nome, estrelas, diaria, cidade, site_id):
self.hotel_id = hotel_id
self.nome = nome
self.estrelas = estrelas
self.diaria = diaria
self.cidade = cidade
self.site_id = site_id
def json(self):
return {
'hotel_id': self.hotel_id,
'nome': self.nome,
'estrelas': self.estrelas,
'diaria': self.diaria,
'cidade': self.cidade,
'site_id': self.site_id
}
@classmethod
def find_hotel(cls, hotel_id):
hotel = cls.query.filter_by(hotel_id=hotel_id).first(
) #SELECT * FROM hoteis where hotel_id = hotel_id limmit 1
if hotel:
return hotel
@classmethod
def find_hotel_by_site_id(cls, site_id):
hotel = cls.query.filter_by(site_id=site_id).first(
) #SELECT * FROM hoteis where hotel_id = hotel_id limmit 1
if hotel:
return hotel
def save_hotel(self):
banco.session.add(self)
banco.session.commit()
def update_hotel(self, nome, estrelas, diaria, cidade):
self.nome = nome
self.estrelas = estrelas
self.diaria = diaria
self.cidade = cidade
def delete_hotel(self):
banco.session.delete(self)
banco.session.commit()
class Funcionario_Model(banco.Model):
__tablename__ = 'funcionarios'
func_id = banco.Column(banco.Integer, primary_key=True)
nome = banco.Column(banco.String(80))
idade = banco.Column(banco.String(20))
cargo_id = banco.Column(banco.Integer, banco.ForeignKey('cargos.cargo_id'))
especi_id = banco.Column(banco.Integer,
banco.ForeignKey('especialidades.especi_id'))
def __init__(self, func_id, nome, idade, cargo_id, especi_id):
self.func_id = func_id
self.nome = nome
self.idade = idade
self.cargo_id = cargo_id
self.especi_id = especi_id
def json(self):
return {
'func_id': self.func_id,
'nome': self.nome,
'idade': self.idade,
'cargo_id': self.cargo_id,
'especi_id': self.especi_id
}
@classmethod
def find_func(cls, func_id):
funcionario = cls.query.filter_by(func_id=func_id).first()
if funcionario:
return funcionario
return None
def save_func(self):
banco.session.add(self)
banco.session.commit()
def update_func(self, nome, idade):
self.nome = nome
self.idade = idade
def delete_func(self):
banco.session.delete(self)
banco.session.commit()
class PedidosModel(banco.Model):
__tablename__ = 'pedidos'
pedido_id = banco.Column(banco.Integer, primary_key=True)
nome = banco.Column(banco.Integer, banco.ForeignKey('roupa.nome'))
cor = banco.Column(banco.Integer, banco.ForeignKey('roupa.cor'))
preco = banco.Column(banco.Integer, banco.ForeignKey('roupa.preco'))
user = banco.Column(banco.Integer, banco.ForeignKey('usuarios.user_id'))
def __init__(self, nome, cor, preco, user_id):
self.nome = nome
self.cor = cor
self.preco = preco
self.user_id = user_id
def json(self):
return {
'self.pedido_id': self.pedido_id,
'nome': self.nome,
'cor': self.cor,
'preco': self.preco,
'user_id': self.user_id
}
@classmethod
def find_pedido(cls, pedido_id):
pedido = cls.query.filter_by(pedido_id=pedido_id).first(
) #SELECT * FROM Roupas Where roupa_id = $roupa_id
if pedido:
return pedido
return None
def save_pedido(self):
banco.session.add(self)
banco.session.commit()
def delete_pedido(self):
# deletando todos as roupas associados ao pedido.
[roupa.delete_roupa() for roupa in self.roupas]
# deletando pedido
banco.session.delete(self)
banco.session.commit()
class HotelModel(banco.Model):
__tablename__ = 'hoteis'
hotel_id = banco.Column(banco.String, primary_key=True)
nome = banco.Column(banco.String(80))
estrelas = banco.Column(banco.Float(precision=1))
diaria = banco.Column(banco.Float(precision=2))
cidade = banco.Column(banco.String(40))
site_id = banco.Column(banco.Integer, banco.ForeignKey('sites.site_id'))
#site = banco.relationship('SiteModel')
# Cria os campos do modelo da classe hotel >>Isso é um objeto<<
def __init__(self, hotel_id, nome, estrelas, diaria, cidade, site_id):
self.hotel_id = hotel_id
self.nome = nome
self.estrelas = estrelas
self.diaria = diaria
self.cidade = cidade
self.site_id = site_id
# Funcão que Transforma um objeto em um fomato DICIONÁRIO que converte/ou é um JSON - automaticamente >>> <<<
def json(self):
return {
'hotel_id': self.hotel_id,
'nome': self.nome,
'estrelas': self.estrelas,
'diaria': self.diaria,
'cidade': self.cidade,
'site_id': self.site_id
}
@classmethod
def find_hotel(cls, hotel_id):
hotel = cls.query.filter_by(hotel_id=hotel_id).first()
if hotel:
return hotel
return None
def save_hotel(self):
banco.session.add(self)
banco.session.commit()
def update_hotel(self, nome, estrelas, diaria, cidade, site_id):
self.nome = nome
self.estrelas = estrelas
self.diaria = diaria
self.cidade = cidade
self.site_id = site_id
def delete_hotel(self):
banco.session.delete(self)
banco.session.commit()
class RotaModel(banco.Model):
__tablename__ = 'rotas'
rota_id = banco.Column(banco.Integer, primary_key=True)
caminhoneiro_id = banco.Column(banco.Integer)
origem = banco.Column(banco.String(80))
destino = banco.Column(banco.String(80))
data_partida = banco.Column(banco.String(80))
tipo_veiculo = banco.Column(banco.Integer,
banco.ForeignKey('veiculos.codigo'))
carregado = banco.Column(banco.String(3))
def __init__(self, caminhoneiro_id, origem, destino, data_partida,
tipo_veiculo, carregado):
self.caminhoneiro_id = caminhoneiro_id
self.origem = origem
self.destino = destino
self.data_partida = data_partida
self.tipo_veiculo = tipo_veiculo
self.carregado = carregado
def json(self):
return {
'rota_id': self.rota_id,
'caminhoneiro_id': self.caminhoneiro_id,
'origem': self.origem,
'destino': self.destino,
'data_partida': self.data_partida,
'tipo_veiculo': self.tipo_veiculo,
'carregado': self.carregado
}
@classmethod
def find_rota(cls, rota_id):
rota = cls.query.filter_by(rota_id=rota_id).first()
if rota:
return rota
return None
def save_rota(self):
banco.session.add(self)
banco.session.commit()
def update_rota(self, caminhoneiro_id, origem, destino, data_partida,
tipo_veiculo, carregado):
self.caminhoneiro_id = caminhoneiro_id
self.origem = origem
self.destino = destino
self.data_partida = datetime.strptime(data_partida, '%Y-%m-%d %H:%M')
self.tipo_veiculo = tipo_veiculo
self.carregado = carregado
def delete_rota(self):
banco.session.delete(self)
banco.session.commit()
class HotelModel(banco.Model):
__tablename__ = 'hotel'
id = banco.Column(banco.String, primary_key=True)
nome = banco.Column(banco.String(80))
estrelas = banco.Column(banco.Float(precision=1))
diaria = banco.Column(banco.Float(precision=2))
cidade = banco.Column(banco.String(40))
site_id = banco.Column(banco.Integer, banco.ForeignKey('site.id'))
def __init__(self, id, nome, estrelas, diaria, cidade, site_id):
self.id = id
self.nome = nome
self.estrelas = estrelas
self.diaria = diaria
self.cidade = cidade
self.site_id = site_id
def json(self):
return {
'id': self.id,
'nome': self.nome,
'estrelas': self.estrelas,
'diaria': self.diaria,
'cidade': self.cidade,
'site_id': self.site_id
}
#cls define que o método é um 'método de classe', portanto não acessa as propriedades self
@classmethod
def find(cls, id):
hotel = cls.query.filter_by(id=id).first()
return hotel if hotel else None
@classmethod
def find_all(cls):
hoteis = cls.query.all()
return hoteis
def save(self):
banco.session.add(self)
banco.session.commit()
def update(self, nome, estrelas, diaria, cidade, site_id):
self.nome = nome
self.estrelas = estrelas
self.diaria = diaria
self.cidade = cidade
self.site_id = site_id
self.save()
def delete(self):
banco.session.delete(self)
banco.session.commit()
class HotelModel(banco.Model):
__tablename__ = 'hoteis'
hotel_id = banco.Column(banco.String, primary_key=True)
nome = banco.Column(banco.String(80))
estrelas = banco.Column(banco.Float(precision=1))
diaria = banco.Column(banco.Float(precision=2))
cidade = banco.Column(banco.String(40))
site_id = banco.Column(banco.Integer, banco.ForeignKey('sites.site_id'))
# site = banco.relationship('SiteModel') #Retorna uma lista de sites?
def __init__(self, hotel_id, nome, estrelas, diaria, cidade, site_id):
self.hotel_id = hotel_id
self.nome = nome
self.estrelas = estrelas
self.diaria = diaria
self.cidade = cidade
self.site_id = site_id
def json(self):
return {
"hotel_id": self.hotel_id,
"nome": self.nome,
"estrelas": self.estrelas,
"diaria": self.diaria,
"cidade": self.cidade,
"site_id": self.site_id
}
@classmethod
def find_hotel(cls, hotel_id):
# SELECT * FROM hoteis WHERE hotel_id=hotel_id
hotel = cls.query.filter_by(hotel_id=hotel_id).first()
if hotel:
return hotel
return None
def save_hotel(self):
banco.session.add(self)
banco.session.commit()
def update_hotel(self, nome, estrelas, diaria, cidade, site_id):
self.nome = nome
self.estrelas = estrelas
self.diaria = diaria
self.cidade = cidade
self.site_id = site_id
def delete_hotel(self):
banco.session.delete(self)
banco.session.commit()
class HotelModel(banco.Model):
__tablename__ = 'hoteis'
hotel_id = banco.Column(banco.String, primary_key=True)
nome = banco.Column(banco.String(80))
estrelas = banco.Column(banco.Float(precision=1))
diaria = banco.Column(banco.Float(precision=2))
cidade = banco.Column(banco.String(80))
site_id = banco.Column(banco.Integer, banco.ForeignKey('sites.site_id'))
# Construtor da classe, que irá retornar um objeto do tipo hotel
def __init__(self, hotel_id, nome, estrelas, diaria, cidade, site_id):
self.hotel_id = hotel_id
self.nome = nome
self.estrelas = estrelas
self.diaria = diaria
self.cidade = cidade
self.site_id = site_id
# Método pega o objeto e o retorna no formato json
def toJson(self):
return {
"hotel_id": self.hotel_id,
"nome": self.nome,
"estrelas": self.estrelas,
"diaria": self.diaria,
"cidade": self.cidade,
"site_id": self.site_id,
}
@classmethod
def find_hotel(cls, hotel_id):
hotel = cls.query.filter_by(hotel_id=hotel_id).first()
if hotel:
return hotel
return None
def save_hotel(self):
banco.session.add(self)
banco.session.commit()
def update_hotel(self, nome, estrelas, diaria, cidade):
self.nome = nome
self.estrelas = estrelas
self.diaria = diaria
self.cidade = cidade
def delete_hotel(self):
banco.session.delete(self)
banco.session.commit()
class HotelModel(banco.Model):
__tablename__ = "hoteis"
hotel_id = banco.Column(banco.String, primary_key=True)
name = banco.Column(banco.String(80))
estrelas = banco.Column(banco.Float(precision=1))
diaria = banco.Column(banco.Float(precision=2))
cidade = banco.Column(banco.String(40))
site_id = banco.Column(banco.Integer, banco.ForeignKey('sites.site_id'))
def __init__(self, hotel_id, name, estrelas, diaria, cidade, site_id):
self.hotel_id = hotel_id
self.name = name
self.estrelas = estrelas
self.diaria = diaria
self.cidade = cidade
self.site_id = site_id
@classmethod
def findHotel(cls, hotel_id):
# SELECT * FROM hoteis WHERE hotel_id = hotel_id
hotel = cls.query.filter_by(hotel_id=hotel_id).first()
if hotel:
return hotel
return None
def save_hotel(self):
banco.session.add(self) #add the self object
banco.session.commit()
def update_hotel(self, name, estrelas, diaria, cidade, site_id):
self.name = name
self.estrelas = estrelas
self.diaria = diaria
self.cidade = cidade
self.site_id = site_id
def delete_hotel(self):
banco.session.delete(self)
banco.session.commit()
def json(self):
return {
'hotel_id': self.hotel_id,
'name': self.name,
'estrelas': self.estrelas,
'diaria': self.diaria,
'cidade': self.cidade,
'site_id': self.site_id
}
class HotelModel(banco.Model):
__tablename__ = 'hoteis'
hotel_id = banco.Column(banco.Integer, primary_key=True)
nome = banco.Column(banco.String(80))
estrelas = banco.Column(banco.Float(precision=1))
diaria = banco.Column(banco.Float(precision=2))
cidade = banco.Column(banco.String(40))
site_id = banco.Column(banco.Integer, banco.ForeignKey('sites.site_id'))
def __init__(self, hotel_id, nome, estrelas, diaria, cidade, site_id):
self.hotel_id = hotel_id
self.nome = nome
self.estrelas = estrelas
self.diaria = diaria
self.cidade = cidade
self.site_id = site_id
@classmethod
def encontrar_hotel(cls, hotel_id):
hotel = cls.query.filter_by(hotel_id=hotel_id).first()
if hotel:
return hotel
return None
def salvar_hotel(self):
banco.session.add(self)
banco.session.commit()
def atualizar_hotel(self, nome, estrelas, diaria, cidade):
self.nome = nome
self.estrelas = estrelas
self.diaria = diaria
self.cidade = cidade
def remover_hotel(self):
banco.session.delete(self)
banco.session.commit()
def json(self):
return {
'hotel_id': self.hotel_id,
'nome': self.nome,
'estrelas': self.estrelas,
'diaria': self.diaria,
'cidade': self.cidade,
'site_id': self.site_id,
}
class HotelModel(banco.Model):
__tablename__ = 'hoteis'
id = banco.Column(banco.String(80), primary_key=True)
nome = banco.Column(banco.String(80))
estrelas = banco.Column(banco.Float(precision=1))
diaria = banco.Column(banco.Float(precision=2))
cidade = banco.Column(banco.String(40))
site_id = banco.Column(banco.Integer, banco.ForeignKey('sites.id'))
def __init__(self, id, nome, estrelas, diaria, cidade, site_id):
self.id = id
self.nome = nome
self.estrelas = estrelas
self.diaria = diaria
self.cidade = cidade
self.site_id = site_id
def json(self):
return {
'id': self.id,
'nome': self.nome,
'estrelas': self.estrelas,
'diaria': self.diaria,
'cidade': self.cidade,
'site_id': self.site_id
}
@classmethod
def find(cls, hotel_id):
hotel = cls.query.filter_by(id=hotel_id).first()
return hotel or None
def save(self):
banco.session.add(self)
banco.session.commit()
def update(self, nome, estrelas, diaria, cidade, site_id):
self.nome = nome
self.estrelas = estrelas
self.diaria = diaria
self.cidade = cidade
self.site_id = site_id
def delete(self):
banco.session.delete(self)
banco.session.commit()
class PointModel(banco.Model):
__tablename__ = 'points'
id = banco.Column(banco.Integer, primary_key=True)
date = banco.Column(banco.String)
hour = banco.Column(banco.String)
employee_id = banco.Column(banco.ForeignKey('employees.id'))
def __init__(self, date, hour, employee_id):
self.date = date
self.hour = hour
self.employee_id = employee_id
def json(self):
return {
'id': self.id,
'date': self.date,
'hour': self.hour,
'employee_id': self.employee_id
}
@classmethod
def find_point(cls, id):
point = cls.query.filter_by(id=id).first()
if point:
return point
return None
@classmethod
def find_day(cls, date):
points = cls.query.filter_by(date=date)
if points:
return points
return None
def save(self):
banco.session.add(self)
banco.session.commit()
def update(self, date, hour):
self.date = date
self.hour = hour
def delete(self):
banco.session.delete(self)
banco.session.commit()
class SiteModel(banco.Model):
__tablename__ = "sites"
id = banco.Column(banco.Integer, primary_key=True, autoincrement=True)
nome_site = banco.Column(banco.String(40))
tpo_site = banco.Column(banco.Integer,
banco.ForeignKey('tipo_site.id'),
nullable=False)
bloqueio = banco.Column(banco.String(10))
def __init__(self, nome_site, tipo_site, bloqueio):
self.nome_site = nome_site
self.tpo_site = tipo_site
self.bloqueio = bloqueio
def json(self):
return {
'id': self.id,
'nome_site': self.nome_site,
'tpo_site': self.tpo_site,
'bloqueio': self.bloqueio
}
@classmethod
def find_all(cls):
site = cls.query.filter_by(bloqueio="S").all()
if site:
return site
return None
@classmethod
def find_by_id(cls, id):
site = cls.query.filter_by(id=id).first()
if site:
return site
return None
@classmethod
def find_by_tipo_site(cls, tipo_site):
sites = cls.query.filter_by(tpo_site=tipo_site, bloqueio="S").all()
return sites
@classmethod
def find_all_bloqu(cls):
sites = cls.query.filter_by(bloqueio="N").all()
if sites:
return sites
return None
@classmethod
def retorna_path(cls, tipo_site):
if tipo_site == 1:
return URLECOMMERCE
elif tipo_site == 2:
return REDESSOCIAIS
elif tipo_site == 3:
return JOGOSONLINE
elif tipo_site == 4:
return SITESPORNOGRAFICOS
def atualiza_txt(self):
tipo_de_site = self.tpo_site
sites = SiteModel.find_by_tipo_site(tipo_de_site)
path = SiteModel.retorna_path(tipo_de_site)
try:
arquivo = open(path, 'w')
for site in sites:
arquivo.write(site.nome_site + "\n")
arquivo.close()
except Exception as ex:
print(str(ex))
def update_site(self, nome_site, tpo_site, bloqueio):
self.nome_site = nome_site
self.tpo_site = tpo_site
self.bloqueio = bloqueio
def save_site(self):
banco.session.add(self)
banco.session.commit()
