def remove_funcionario(self, funcionario: Funcionario):
if not isinstance(funcionario, Funcionario):
raise self.NotSocioException('Apenas funcionários podem ser adicionados a ramais.')
if isinstance(funcionario, Socio):
raise self.NotSocioException('Sócios não podem ser adicionados a ramais.')
c.execute("DELETE FROM Ramal WHERE funcionario_id = ? AND reserva_id = ?",
(funcionario.id, self.__id,))
conn.commit()
def create(cls, socio: 'Socio', sala: 'Sala', horario: 'Horario'):
if not isinstance(socio, Socio):
raise cls.NotSocioException('É necessário ser um sócio para fazer uma reserva.')
# check if another reserva with same socio/sala/horario parameters already exists.
c.execute(
"SELECT id FROM Reserva WHERE (socio_id = ? OR sala_id = ?) AND horario = ?",
(socio.id, sala.id, str(horario),)
)
if c.fetchone():
raise cls.AlreadyExistsException(
"Uma reserva já foi feita nesta combinação de sócio/sala e horário.")
c.execute("INSERT INTO Reserva (socio_id, sala_id, horario) VALUES (?,?,?)",
(socio.id, sala.id, str(horario),))
this_id = c.lastrowid
conn.commit()
return Reserva(this_id, socio, sala, horario)
def add_funcionario(self, funcionario: Funcionario):
if not isinstance(funcionario, Funcionario):
raise self.NotSocioException('Apenas funcionários podem ser adicionados a ramais.')
if isinstance(funcionario, Socio):
raise self.NotSocioException('Sócios não podem ser adicionados a ramais.')
c.execute("SELECT funcionario_id FROM Ramal WHERE funcionario_id = ? AND reserva_id = ?",
(funcionario.id, self.__id,))
if c.fetchone():
raise self.AlreadyExistsException("O funcionário em questão já possui um Ramal nesta reserva.")
c.execute("SELECT reserva_id FROM Ramal WHERE reserva_id = ?", (self.__id,))
actual_length = len(c.fetchall())
if actual_length >= self.sala.vagas:
raise Sala.ExceedingLengthException('A quantidade de vagas na sala acabou.')
c.execute("INSERT INTO Ramal (funcionario_id, reserva_id) VALUES (?,?)",
(funcionario.id, self.id,))
conn.commit()
def delete(self):
c.execute("DELETE FROM Funcionario WHERE id = ?", (self.id, ))
conn.commit()
c.execute("DELETE FROM Reserva WHERE socio_id IS NULL")
conn.commit()
c.execute("DELETE FROM Ramal WHERE reserva_id IS NULL")
conn.commit()
def delete(self):
c.execute("DELETE FROM Sala WHERE id = ?", (self.__id, ))
conn.commit()
c.execute("DELETE FROM Reserva WHERE sala_id IS NULL")
conn.commit()
c.execute("DELETE FROM Ramal WHERE reserva_id IS NULL")
conn.commit()
def horario(self, value: 'Horario'):
c.execute("UPDATE Reserva SET horario = ? WHERE id = ?", (str(value), self.__id,))
conn.commit()
self.__horario = value
def socio(self, value: 'Socio'):
socio_id = value.id
c.execute("UPDATE Reserva SET socio_id = ? WHERE id = ?", (socio_id, self.__id,))
conn.commit()
self.__socio = value
def create(cls, name: str):
c.execute("INSERT INTO Funcionario (name, role) VALUES (?,?)",
(name, cls.get_role(),))
this_id = c.lastrowid
conn.commit()
return cls(name, this_id)
def create(cls, vagas: int):
c.execute("INSERT INTO Sala (spaces) VALUES (?)", (vagas, ))
this_id = c.lastrowid
conn.commit()
return Sala(this_id, vagas)
