fix 36

tmessage/cli.py

@@ -4,7 +4,7 @@
 import paho.mqtt.client as mqtt
 from colorama import init, deinit, Fore, Back, Style
 import tmessage.auth as auth  # auth.py
-from tmessage.db import store_messages  # db.py
+from tmessage.db import grab_messages, store_messages  # db.py"""
 
 
 # Initialize colorama
@@ -80,6 +80,8 @@ def main():
             pub_msg = f'[{user_name}] {displayed_name}: {raw_msg}'
             if raw_msg != '':
                 MQTT_CLIENT.publish(MQTT_TOPIC, pub_msg)
+                if raw_msg == "grab":
+                    grab_messages(CURRENT_USER)
                 if IS_STORE:
                     store_messages(CURRENT_USER, raw_msg)
             else:

tmessage/db.py

@@ -16,12 +16,17 @@ class Message(Model):
     class Meta:  # pylint: disable=missing-class-docstring,too-few-public-methods
         database = MESSAGES_DB
 
-
+@MESSAGES_DB
+def grab_messages(user):
+    """Grab messages from the user"""
+    cursor = MESSAGES_DB.execute_sql("SELECT sender, message FROM message where sender LIKE \'" + user +'\'')
+    for row in cursor.fetchall():
+       print(row[0] + " : " + row[1])
+
 @MESSAGES_DB
 def store_messages(user, raw_msg):
     """Store a message sent by the indicated user in the database"""
     time = datetime.now()
-
     Message.create(sender=user, message=raw_msg, timestamp=time)