def carregarMensagems(self, usuario):
mensagemDAO = MensagemDAO()
usuario.mensagems = []
mensagems = mensagemDAO.getListMensagemUser(usuario)
for mensagem in mensagems:
mensagemUser = Mensagem()
mensagemUser.id = mensagem[0]
mensagemUser.id_user_envia = mensagem[1]
mensagemUser.id_user_recebe = mensagem[2]
mensagemUser.mensagem = mensagem[3]
mensagemUser.data_envio = mensagem[4]
usuario.mensagems.append(
mensagemUser
) # Adicionando a mensagem a lista de mensagems do usuário
def criarMensAmigo(self, usuario):
if usuario.totalAmigos() == 0:
print("\nVocê não possue amigos\n")
else:
mensagemDAO = MensagemDAO()
amigo = Amigo()
amigo.verAmigosNome(
usuario
) # Listando amigos para o usuário escolher para quem vai enviar a mensagem
print(
"OBS: Ao digitar zero(0) a operação é cancelada e você retorna para o menu de Mensagems\n"
)
print("--------------------------------------------------------")
print("Digite o número do amigo que deseja enviar a mensagem")
print("--------------------------------------------------------")
while (True):
try:
numAmigo = int(
input(">>:")) # Armazenando a escolha do usuário
if (numAmigo > usuario.totalAmigos() or numAmigo < 0):
print(
"\nVocê não tem nenhum amigo atribuido a este número\n"
)
break # Encerrando loop
elif numAmigo == 0:
break
else:
self.id_user_envia = usuario.id
self.id_user_recebe = usuario.amigos[
numAmigo -
1].id # Atribuindo o ID do amigo que o usuário escolheu
self.data_envio = datetime.datetime.today()
self.mensagem = input("Mensagem: ")
mensagemDAO.enviarMensagem(self)
print("\nMensagem enviada\n")
break
except ValueError:
print(numeroInvalido)
def criarMensDesconhecido(self, usuario):
usuarioDAO = UsuarioDAO()
mensagemDAO = MensagemDAO()
print(
"\nPara enviar uma mensagem para um usuário que não é seu amigo você precissa informar o E-mail dele\n"
)
email = input("E-mail do usuário: ")
exist = usuarioDAO.usuarioExist(email)
if exist:
self.id_user_envia = usuario.id
self.id_user_recebe = usuarioDAO.getIdUsuario(email)
self.data_envio = datetime.datetime.today()
self.mensagem = input("\nMensagem: ")
if mensagemDAO.enviarMensagem(
self): # Se a mensagem for enviada a função retorna True
print("\nMensagem enviada\n")
else:
print(
"\nNão existe nenhum usuário com este E-mail na rede Nicks\n")
def atualizar(self):
MensagemDAO.alterar()
def deletar(self):
MensagemDAO.deletar()
def inserir(self):
MensagemDAO.inserir()
def listar(self):
MensagemDAO.listar()
def menuDeUsuario(usuario: Usuario):
while True:
try:
opcao = int(input(
"\n1) Ver perfil\n" \
"2) Add Amigo\n" \
"3) Mandar mensagem\n" \
"4) Ver amigos\n" \
"5) Ver mensagens\n" \
"0) Sair\n\n" \
"->"
))
except:
print("Opção inválida.")
continue
if(opcao == 0):
break
elif(opcao == 1):
print("\nNome: %s" \
"\nE-mail: %s" \
"\nSexo: %s" \
"\nCidade: %s" \
"\nData de nascimento: %s" % (usuario.nome, usuario.email, usuario.sexo, usuario.cidade, usuario.data_nascimento))
elif(opcao == 2):
nome = input("Nome: ")
usuarios = UsuarioDAO().procurarPeloNome(nome)
if(len(usuarios) == 0):
print("Nenhum usuário com este nome encontrado.")
continue
elif(len(usuarios) == 1):
try:
UsuarioDAO().addAmigo(usuario.id, usuarios[0].id)
print("Amigo adicionado.")
except Exception as err:
print(err)
else:
print()
i = 1
for u in usuarios:
print("%s) %s - %s" % (i, u.nome, u.email))
i += 1
n = input("Digite o número do usuário: ")
u = UsuarioDAO().procurarPeloId(usuarios[u - 1].id)
if(u is None):
print("Usuário inválido.")
else:
UsuarioDAO().addAmigo(usuario.id, u.id)
print("%s foi adicionado aos seus amigos." % (u.nome))
elif(opcao == 3):
nome = input("Nome: ")
usuarios = UsuarioDAO().procurarPeloNome(nome)
if (len(usuarios) == 0):
print("Nenhum usuário com este nome encontrado.")
continue
elif (len(usuarios) == 1):
try:
texto = input("Mensagem: ")
mensagem = Mensagem(usuario, usuarios[0], texto, str(date.today()))
MensagemDAO().insert(mensagem)
print("Mensagem enviada.")
except Exception as err:
print(err)
else:
print()
i = 1
for u in usuarios:
print("%s) %s - %s" % (i, u.nome, u.email))
i += 1
n = input("Digite o número do usuário: ")
u = UsuarioDAO().procurarPeloId(usuarios[u - 1].id)
if (u is None):
print("Usuário inválido.")
else:
try:
texto = input("Mensagem: ")
mensagem = Mensagem(usuario, usuarios[0], texto, str(date.today()))
MensagemDAO().insert(mensagem)
print("Mensagem enviada.")
except Exception as err:
print(err)
elif(opcao == 4):
amigos = UsuarioDAO().listarAmigos(usuario.id)
i = 1
for amigo in amigos:
print("%s) %s" % (i, amigo.nome))
elif(opcao == 5):
mensagens = MensagemDAO().listarDeUsuario(usuario.id)
for mensagem in mensagens:
print("(%s) %s -> %s: %s" % (mensagem.data_envio, mensagem.remetente.nome, mensagem.destinatario.nome, mensagem.texto))