def UpdateSeats(self, Model, seatClass):
Database = DB()
Database.run(
"UPDATE Seats SET modelCode = %s, seatClass = %s WHERE seatNumber = %s ;",
(str(Model.code), seatClass, str(self.seatNumber)))
self.Model = Model
self.seatClass = seatClass
def InsertUser(self, name, lastname, mail, password, administrador=0):
Database = DB()
userCursor = Database.run(
"INSERT INTO User VALUES(NULL, %s, %s, %s, %s, %s)",
(str(name), str(lastname), str(mail), str(password),
administrador))
self.idUser = userCursor.lastrowid
def UpdatePlane(self, Model, constructionDay):
Database = DB()
Database.run(
"UPDATE Plane SET modelCode = %s, constructionDay = %s WHERE idPlane = %s;",
(str(Model.code), str(constructionDay), str(self.idPlane)))
self.Model = Model
self.constructionDay = constructionDay
def SelectPlaneID(idPlane):
Database = DB()
planeCursor = Database.run("SELECT * FROM Plane WHERE idPlane = %s;",
str(idPlane))
planeDict = planeCursor.fetchone()
tmpPlane = Plane.GetPlane(planeDict)
plane = Plane(tmpPlane[0], tmpPlane[1], tmpPlane[2])
return plane
def InsertFlight(self):
Database = DB()
flightCursor = Database.run(
"INSERT INTO Flight VALUES(NULL, %s, %s, %s, %s, %s, %s);",
(self.departure, self.arrival, str(
self.Plane.idPlane), self.flightDepartureDatetime,
self.flightArrivalDatetime, self.percentDiscount))
self.idFlight = flightCursor.lastrowid
def SelectSeatsID(seatNumber):
Database = DB()
seatsCursor = Database.run(
"SELECT * FROM Seats WHERE seatNumber = %s;", str(seatNumber))
seatsDict = seatsCursor.fetchone()
tmpSeat = Seats.GetSeat(seatsDict)
Seat = Seats(tmpSeat[0], tmpSeat[1], tmpSeat[2])
return Seat
def SelectPlaneModelsID(modelCode):
Database = DB()
planeModelCursor = Database.run(
"SELECT * FROM PlaneModel WHERE code = %s;", str(modelCode))
planeModelDict = planeModelCursor.fetchone()
model = PlaneModel.GetPlaneModel(planeModelDict)
Model = PlaneModel(model[0], model[1])
return Model
def SelectFlightsID(idFlight):
Database = DB()
flightsCursor = Database.run(
"SELECT * FROM Flight WHERE idFlight = %s;", str(idFlight))
flightsDict = flightsCursor.fetchone()
tmpFlight = Flight.GetFlight(flightsDict)
flight = Flight(tmpFlight[0], tmpFlight[1], tmpFlight[2], tmpFlight[3],
tmpFlight[4], tmpFlight[5], tmpFlight[6])
return flight
def SelectUserID(idUser):
Database = DB()
userCursor = Database.run("SELECT * FROM User WHERE idUser = %s;",
str(idUser))
userDict = userCursor.fetchone()
tmpUser = User.GetUser(userDict)
user = User(tmpUser[0], tmpUser[1], tmpUser[2], tmpUser[3], tmpUser[4],
tmpUser[5])
return user
def GetFlight(dic):
Database = DB()
planeCursor = Database.run("SELECT * FROM Plane WHERE idPlane = %s;",
str(dic["idPlane"]))
planeDict = planeCursor.fetchone()
tmpPlane = ClassPlane.GetPlane(planeDict)
Plane = ClassPlane(tmpPlane[0], tmpPlane[1], tmpPlane[2])
return dic["idFlight"], dic["departure"], dic["arrival"], Plane, dic["flightDepartureDatetime"],\
dic["flightArrivalDatetime"], dic["percentDiscount"]
def SelectPlanes():
Database = DB()
planeCursor = Database.run("SELECT * FROM Plane;")
planeDict = planeCursor.fetchall()
planesList = []
for item in planeDict:
tmpPlane = Plane.GetPlane(item)
plane = Plane(tmpPlane[0], tmpPlane[1], tmpPlane[2])
planesList.append(plane)
return planesList
def SelectPlaneModels():
Database = DB()
planeModelCursor = Database.run("SELECT * FROM PlaneModel;")
planeModelDict = planeModelCursor.fetchall()
planeModelList = []
for item in planeModelDict:
tmpModel = PlaneModel.GetPlaneModel(item)
Model = PlaneModel(tmpModel[0], tmpModel[1])
planeModelList.append(Model)
return planeModelList
def SelectFlightUserID(idFlight, idUser):
Database = DB()
cursorFlightUser = Database.run(
"SELECT * FROM Flight_has_User WHERE idFlight = %s AND idUser = %s;",
(str(idFlight), str(idUser)))
dictFlightUser = cursorFlightUser.fetchone()
tmpFlightUser = FlightUser.GetFlightUser(dictFlightUser)
flightUser = FlightUser(tmpFlightUser[0], tmpFlightUser[1],
tmpFlightUser[2])
return flightUser
def SelectSeats():
Database = DB()
seatsCursor = Database.run("SELECT * FROM Seats;")
seatsDict = seatsCursor.fetchall()
seatsList = []
for item in seatsDict:
tmpSeat = Seats.GetSeat(item)
Seat = Seats(tmpSeat[0], tmpSeat[1], tmpSeat[2])
seatsList.append(Seat)
return seatsList
def SelectFlightUser():
Database = DB()
cursorFlightUser = Database.run("SELECT * FROM Flight_has_User;")
dictFlightUser = cursorFlightUser.fetchall()
listaFlightUser = []
for item in dictFlightUser:
tmpFlightUser = FlightUser.GetFlightUser(item)
flightUser = FlightUser(tmpFlightUser[0], tmpFlightUser[1],
tmpFlightUser[2])
listaFlightUser.append(flightUser)
return listaFlightUser
def UpdateFlightUser(self, flightObj, userObj, seatObj):
Database = DB()
Database.run(
"UPDATE Flight_has_User SET seatNumber = %s, idFlight= %s, idUser= %s WHERE idFlight = %s"
"AND idUser = %s;",
(str(seatObj.seatNumber), str(flightObj.idFlight),
str(userObj.idUser), str(
self.FlightObj.idFlight), str(self.UserObj.idUser)))
self.FlightObj = flightObj
self.UserObj = userObj
self.Seat = seatObj
def SelectUser():
Database = DB()
userCursor = Database.run("SELECT * FROM User;")
userDict = userCursor.fetchall()
userLista = []
for item in userDict:
tmpUser = User.GetUser(item)
user = User(tmpUser[0], tmpUser[1], tmpUser[2], tmpUser[3],
tmpUser[4], tmpUser[5])
userLista.append(user)
return userLista
def SelectUsers(idUser):
Database = DB()
cursorFlightUser = Database.run(
"SELECT * FROM Flight_has_User WHERE idUser = %s;", str(idUser))
dictFlightUser = cursorFlightUser.fetchall()
listaVuelosUser = []
for item in dictFlightUser:
tmpFlightUser = FlightUser.GetFlightUser(item)
flightUser = FlightUser(tmpFlightUser[0], tmpFlightUser[1],
tmpFlightUser[2])
listaVuelosUser.append(flightUser)
return listaVuelosUser
def SelectUserMailPassword(mail, password):
Database = DB()
userCursor = Database.run(
"SELECT * FROM User WHERE mail = %s AND password = %s;",
(str(mail), str(password)))
userDict = userCursor.fetchone()
if userDict is None:
return None
tmpUser = User.GetUser(userDict)
user = User(tmpUser[0], tmpUser[1], tmpUser[2], tmpUser[3], tmpUser[4],
tmpUser[5])
return user
def SelectFlights():
Database = DB()
flightsCursor = Database.run("SELECT * FROM Flight;")
flightsDict = flightsCursor.fetchall()
flightList = []
for item in flightsDict:
tmpFlight = Flight.GetFlight(item)
flight = Flight(tmpFlight[0], tmpFlight[1], tmpFlight[2],
tmpFlight[3], tmpFlight[4], tmpFlight[5],
tmpFlight[6])
flightList.append(flight)
return flightList
def UpdateFlight(self, departure, arrival, Plane, flightDepartureDatetime,
flightArrivalDatetime, percentDiscount):
Database = DB()
Database.run(
"UPDATE Flight SET departure = %s, arrival = %s, idPlane = %s, flightDepartureDatetime = %s,"
"flightArrivalDatetime = %s WHERE idFlight = %s;",
(departure, arrival, str(Plane.idPlane), str(
self.idFlight), str(flightDepartureDatetime),
str(flightArrivalDatetime), str(percentDiscount)))
self.departure = departure
self.arrival = arrival
self.Plane = Plane
self.flightDepartureDatetime = flightDepartureDatetime
self.flightArrivalDatetime = flightArrivalDatetime
def BuscarSalida(salida):
Database = DB()
flightsCursor = Database.run(
"SELECT * FROM Flight WHERE departure = %s", (str(salida)))
flightsDict = flightsCursor.fetchall()
flightList = []
if flightsDict is not None:
for item in flightsDict:
tmpFlight = Flight.GetFlight(item)
flight = Flight(tmpFlight[0], tmpFlight[1], tmpFlight[2],
tmpFlight[3], tmpFlight[4], tmpFlight[5],
tmpFlight[6])
flightList.append(flight)
return flightList
return None
def BuscarViaje(salida, llegada, fechaIda, fechaVuelta):
Database = DB()
flightsCursor = Database.run(
"SELECT * FROM Flight WHERE departure = %s and arrival = %s and "
"flightDepartureDatetime = %s and flightArrivalDatetime = %s;",
(str(salida), str(llegada),
str(datetime.strptime(fechaIda, '%Y-%m-%d %H:%M:%S')),
str(datetime.strptime(fechaVuelta, '%Y-%m-%d %H:%M:%S'))))
flightsDict = flightsCursor.fetchone()
if flightsDict is not None:
tmpFlight = Flight.GetFlight(flightsDict)
flight = Flight(tmpFlight[0], tmpFlight[1], tmpFlight[2],
tmpFlight[3], tmpFlight[4], tmpFlight[5],
tmpFlight[6])
return flight
return None
def BuscarSalidaLLegadaFechaIda(salida, llegada, fechaIda):
Database = DB()
flightsCursor = Database.run(
"SELECT * FROM Flight WHERE departure = %s and arrival = %s and "
"flightDepartureDatetime = %s",
(str(salida), str(llegada),
str(datetime.strptime(fechaIda, '%Y-%m-%d %H:%M:%S'))))
flightsDict = flightsCursor.fetchall()
flightList = []
if flightsDict is not None:
for item in flightsDict:
tmpFlight = Flight.GetFlight(item)
flight = Flight(tmpFlight[0], tmpFlight[1], tmpFlight[2],
tmpFlight[3], tmpFlight[4], tmpFlight[5],
tmpFlight[6])
flightList.append(flight)
return flightList
return None
from SQLConnection import DB
from flask import Flask, render_template, request, redirect, session
from ClassUser import User
from ClassPlaneModel import PlaneModel
from ClassSeats import Seats
from ClassPlane import Plane
from ClassFlight import Flight
from ClassFlightUser import FlightUser
app = Flask(__name__)
app.secret_key = 'AlusesKey'
Database = DB()
Database.SetConnection("localhost", "root", "alumno", "Aluses")
# user = User('Sebastian', 'Elustondo', '*****@*****.**', 'Patuco20', 1) # ya esta creado, solo para fijarse nombre y contra
# user.InsertUser('Sebastian', 'Elustondo', '*****@*****.**', 'Patuco20', 1)
def Session():
if not 'idUser' in session:
session['idUser'] = session.get('idUser')
@app.route('/home', methods=['GET'])
def home():
active = False
user = User('', '', '', '', '')
if 'idUser' in session:
active = True
user = User.SelectUserID(session['idUser'])
def InsertFlightUser(self):
Database = DB()
Database.run("INSERT INTO Flight_has_User VALUES(%s, %s, %s);",
(str(self.FlightObj.idFlight), str(
self.UserObj.idUser), str(self.Seat.seatNumber)))
def DeleteFlightUser(idFlight, idUser):
Database = DB()
Database.run(
"DELETE FROM Flight_has_User WHERE idFlight = %s AND idUser = %s;",
(str(idFlight), str(idUser)))
def UpdateUser(self, name, lastname, mail, password):
Database = DB()
Database.run(
"UPDATE User SET name = %s, lastname = %s, mail = %s, password = %s WHERE idUser = %s;",
(str(name), str(lastname), str(mail), str(password),
str(self.idUser)))
def DeleteUser(idUser):
Database = DB()
User.DeleteUserFlight(idUser)
Database.run("DELETE FROM User WHERE idUser = %s;", str(idUser))
def DeleteUserFlight(idUser):
Database = DB()
Database.run("DELETE FROM Flight_has_User WHERE idUser = %s;",
str(idUser))
