def setUp(self):
'''Crea una colección para hacer las pruebas'''
self.cuentas_claras = Cuentas_claras()
'''Abre la sesión'''
self.session = Session()
self.data_factory = Faker()
Faker.seed(1000)
# crear nombres y apellidos random invalidos
self.nombres_invalidos= []
for _ in range(10):
self.nombres_invalidos.append(self.data_factory.bothify(text='#?#?##?',letters='!"#$%&/()='))
# crear viajeros random
self.viajeros = []
for _ in range(10):
viajero = Viajero(self.data_factory.unique.first_name(),self.data_factory.unique.last_name())
self.viajeros.append(viajero)
self.session.close()
def setUp(self):
'''Crea una colección para hacer las pruebas'''
self.cuentas_claras = Cuentas_claras()
'''Abre la sesión'''
self.session = Session()
'''Crear Faker '''
self.data_factory = Faker()
Faker.seed(1000)
'''Crea las actividades'''
# crear actividades random
self.actividades = []
for _ in range(1):
actividad = Actividad(nombre=self.data_factory.unique.name())
self.session.add(actividad)
self.actividades.append(actividad)
# crear viajeros random
self.viajeros = []
for _ in range(4):
viajero = Viajero(self.data_factory.unique.first_name(),
self.data_factory.unique.last_name())
self.session.add(viajero)
self.viajeros.append(viajero)
for i in range(len(self.viajeros)):
self.actividades[0].agregar_viajero_en_actividad(self.viajeros[i])
# crear gastos random
self.gastos = []
for _ in range(8):
gasto = Gasto(concepto=self.data_factory.unique.name(),
valor=random.uniform(1, 1000),
fecha_gasto=datetime.strptime(
self.data_factory.date(), '%Y-%m-%d'))
self.session.add(gasto)
self.gastos.append(gasto)
# se asigna aleatoriamente los gastos a los viajeros
for i in range(len(self.gastos)):
viajero_id = random.randrange(len(self.viajeros))
self.viajeros[viajero_id].agregar_gasto(self.gastos[i])
for i in range(len(self.gastos)):
self.actividades[0].agregar_gasto(self.gastos[i])
'''Persiste los objetos y cierra la sesión'''
self.session.commit()
def setUp(self):
'''Crea una colección para hacer las pruebas'''
self.cuentas_claras = Cuentas_claras()
'''Abre la sesión'''
self.session = Session()
self.data_factory = Faker()
Faker.seed(1000)
# crear actividades random
self.actividades = []
for _ in range(2):
actividad = Actividad(nombre=self.data_factory.text())
self.session.add(actividad)
self.actividades.append(actividad)
# crear viajeros random
self.viajeros = []
for _ in range(2):
viajero = Viajero(self.data_factory.unique.first_name(),
self.data_factory.unique.last_name())
self.session.add(viajero)
self.viajeros.append(viajero)
self.actividades[1].agregar_viajero_en_actividad(self.viajeros[0])
self.actividades[1].agregar_viajero_en_actividad(self.viajeros[1])
# crear gastos random
self.gastos = []
for _ in range(2):
gasto = Gasto(concepto=self.data_factory.text(),
valor=random.uniform(1, 1000),
fecha_gasto=datetime.strptime(
self.data_factory.date(), '%Y-%m-%d'))
self.session.add(gasto)
self.gastos.append(gasto)
self.viajeros[0].agregar_gasto(self.gastos[0])
self.viajeros[0].agregar_gasto(self.gastos[1])
self.actividades[1].agregar_gasto(self.gastos[0])
self.actividades[1].agregar_gasto(self.gastos[1])
self.session.commit()
def agregar_viajero(self, nombre, apellido):
if nombre == "" or apellido == "":
return None
if not nombre.isalpha() or not apellido.isalpha():
return None
viajeros = self.session.query(Viajero).filter(
Viajero.nombre == nombre, Viajero.apellido == apellido).all()
if len(viajeros) > 0:
return None
viajero = Viajero(nombre, apellido)
self.session.add(viajero)
self.session.commit()
return True
def setUp(self):
''' Se puebla la base de datos para las pruebas'''
self.session = Session()
busqueda_actividad = self.session.query(Actividad).all()
for actividad in busqueda_actividad:
self.session.delete(actividad)
busqueda_viajero = self.session.query(Viajero).all()
for viajero in busqueda_viajero:
self.session.delete(viajero)
busqueda_gasto = self.session.query(Gasto).all()
for gasto in busqueda_gasto:
self.session.delete(gasto)
busqueda_av = self.session.query(ActividadViajero).all()
for av in busqueda_av:
self.session.delete(av)
self.session.commit()
# Crear actividades y agregarlas a la base de datos con add y commit
self.actividad_1 = Actividad("Paseo a la Playa")
self.actividad_2 = Actividad("Caminata")
self.session.add(self.actividad_1)
self.session.add(self.actividad_2)
self.session.commit()
# Crear los viajeros y agregarlos a la base de datos con add y commit
self.viajero_1 = Viajero("Juan", "Perez")
self.viajero_2 = Viajero("Rosa", "Garcia")
self.viajero_3 = Viajero("Luis", "Mora")
self.viajero_4 = Viajero("Ana", "Zavala")
self.session.add(self.viajero_1)
self.session.add(self.viajero_2)
self.session.add(self.viajero_3)
self.session.add(self.viajero_4)
self.session.commit()
# Crear los gastos. Note que el atributo fecha en este momento es String.
self.gasto_1 = Gasto("Transporte", 100, '21-01-2021')
self.gasto_2 = Gasto("Comida", 200, '22-01-2021')
self.gasto_3 = Gasto("Bebida", 90, '23-01-2021')
self.gasto_4 = Gasto("Fiesta", 120, '24-01-2021')
self.gasto_5 = Gasto("Transporte", 150, '28-01-2021')
self.session.add(self.gasto_1)
self.session.add(self.gasto_2)
self.session.add(self.gasto_3)
self.session.add(self.gasto_4)
self.session.add(self.gasto_5)
self.session.commit()
# Crea las relaciones entre actividad y viajero (cuales son los viajeros asociados a cada actividad)
self.actividad_1.viajeros = [
self.viajero_1, self.viajero_2, self.viajero_3, self.viajero_4
]
self.actividad_2.viajeros = [self.viajero_2]
# Crea las relaciones entre actividad y gasto *(cuales son los gastos de cada actividad)
self.actividad_1.gastos = [
self.gasto_1, self.gasto_2, self.gasto_3, self.gasto_4
]
self.actividad_2.gastos = [self.gasto_5]
# Crea las relaciones entre viajero y gastos (que viajero realizo cada gasto)
self.viajero_1.gastos = [self.gasto_1, self.gasto_3]
self.viajero_2.gastos = [self.gasto_2, self.gasto_5]
self.viajero_3.gastos = [self.gasto_4]
self.viajero_4.gastos = []
self.session.commit()
# Cierra la sesion
self.session.close()
from src.modelo.declarative_base import Session, engine, Base
if __name__ == '__main__':
Base.metadata.create_all(engine)
session = Session()
# Crear actividades y agregarlas a la base de datos con add y commit
actividad_1 = Actividad("Paseo a la Playa")
actividad_2 = Actividad("Caminata")
session.add(actividad_1)
session.add(actividad_2)
session.commit()
# Crear los viajeros y agregarlos a la base de datos con add y commit
viajero_1 = Viajero("Juan", "Perez")
viajero_2 = Viajero("Rosa", "Garcia")
viajero_3 = Viajero("Luis", "Mora")
viajero_4 = Viajero("Ana", "Zavala")
session.add(viajero_1)
session.add(viajero_2)
session.add(viajero_3)
session.add(viajero_4)
session.commit()
# Crear los gastos. Note que el atributo fecha en este momento es String.
gasto_1 = Gasto("Transporte", 100, '21-01-2021')
gasto_2 = Gasto("Comida", 200, '22-01-2021')
gasto_3 = Gasto("Bebida", 90, '23-01-2021')
gasto_4 = Gasto("Fiesta", 120, '24-01-2021')