def find_by_recipient(cursor, recipient_username):
sql = "SELECT id, from_user, to_user, context, created_at FROM messages WHERE to_user=%s"
ret = []
cursor.execute(sql, (recipient_username, ))
for row in cursor.fetchall():
loaded_message = Message()
loaded_message._id = row[0]
loaded_message.from_user = row[1]
loaded_message.to_user = row[2]
loaded_message.context = row[3]
loaded_message.created_at = row[4]
ret.append(loaded_message)
return ret
def load_all_msg(cursor):
sql = "SELECT * FROM messages"
ret = []
cursor.execute(sql)
for row in cursor.fetchall():
loaded_msg = Message()
loaded_msg._id = row[0]
from_user_temp = UserService.find_by_id(cursor, row[1])
loaded_msg.from_user = from_user_temp.username
to_user_temp = UserService.find_by_id(cursor, row[2])
loaded_msg.to_user = to_user_temp.username
loaded_msg.context = row[3]
loaded_msg.created_at = row[4]
ret.append(loaded_msg)
return ret
def chat_submit(request):
context = {}
if "login_user" in request.session:
context["login_user"] = request.session["login_user"]
context["username"] = request.session["login_user"].split("@")[0]
else:
return redirect("/")
if request.POST:
message = Message()
message.email = context["login_user"]
message.content = request.POST["content"]
message.sender = True
message.created_at = timestamp()
messageRepo = MessageRepo()
messageRepo.save(message)
return redirect("/account/viewerdash/chat/")
def find_by_recipient(cursor, recipient_username):
sql = "SELECT id, from_user, to_user, context, created_at FROM messages WHERE to_user=%s"
ret = []
cursor.execute(sql, (recipient_username, ))
data = cursor.fetchall()
if data:
for row in data:
loaded_msg = Message()
loaded_msg._id = row[0]
from_user_temp = UserService.find_by_id(cursor, row[1])
loaded_msg.from_user = from_user_temp.username
to_user_temp = UserService.find_by_id(cursor, row[2])
loaded_msg.to_user = to_user_temp.username
loaded_msg.context = row[3]
loaded_msg.created_at = row[4]
ret.append(loaded_msg)
return ret
else:
return None
def fetch_messages(self, email):
query = "SELECT content, sender, created_at FROM messages WHERE user_email = %s ORDER BY created_at"
try:
with connection.cursor() as cursor:
cursor.execute(query, [email])
rows = cursor.fetchall()
if rows is None:
return None
else:
messages = list()
for row in rows:
message = Message()
message.email = email
message.content = row[0]
message.sender = row[1]
message.created_at = row[2]
messages.append(message)
return messages
except Exception as e:
traceback.print_exc()
return None
user = MessageService.find_by_recipient(cursor, args.username)
print('Wrong data provided')
if args.send == True:
user = UserService.find_by_username(cursor, args.username)
if user is not None:
is_user_logged = UserService.login(cursor, args.username,
args.password)
if is_user_logged == True:
to_user = UserService.find_by_username(cursor, args.username)
if to_user is not None:
context = args.message
if context:
message = Message()
message.from_user = user.id
message.to_user = to_user.id
message.context = args.message
message.created_at = datetime.now()
message.save(cursor)
print("Message successfully sent")
else:
print("Message is empty!")
else:
print("Wrong username")
else:
print('Please login')
else:
print('Invalid username or password')
cursor.close()
connection.close()
print('Username or password invalid')
if args.send == True:
user = UserService.find_by_username(cursor, args.username)
if user is not None:
is_user_logged = UserService.login(cursor, args.username, args.password)
if is_user_logged == True:
to_user = UserService.find_by_email(cursor, args.email)
if to_user is not None:
context = args.message
if context:
msg = Message()
msg.from_user = user.id
msg.to_user = to_user.id
msg.context = args.message
msg.created_at = datetime.now()
msg.save(cursor)
print("Message send")
else:
print("No message provided")
else:
print("No user with that email")
else:
print('Username or password invalid')
else:
print('Username or password invalid')
cursor.close()
connection.close()
