fixed some things

2 files changed, 14 insertions, 4 deletions

diff --git a/lib/commands.py b/lib/commands.py
index ec44558..defdef6 100644
--- a/lib/commands.py
+++ b/lib/commands.py
@@ -20,7 +20,7 @@ class Commands(object):
 
                     self.user = cmd[2]
                     with open(co['path'], 'wb') as fout:
-                        fout.write(cmd[2].encode('utf-8'))
+                        fout.write((cmd[2] + '\n').encode('utf-8'))
 
                     print('Ok')
                 elif cmd[1] is not ListCommands.commands['set']:
@@ -82,7 +82,7 @@ class Commands(object):
 
                 try:
                     with open(co['path'], 'ab') as fout:
-                        fout.write(('\n' + msg).encode('utf-8'))
+                        fout.write((msg + '\n').encode('utf-8'))
 
                     print('Ok')
                 except:
diff --git a/lib/web.py b/lib/web.py
index 69522f8..bbad99d 100644
--- a/lib/web.py
+++ b/lib/web.py
@@ -1,11 +1,21 @@
 from flask import Flask, render_template
-from .classes import r
+from .config import config
 
 app = Flask(__name__)
 
 @app.route("/")
 def hello():
-    lista = r.zrange('todo', 0, -1)
+    with open(config['path'], 'rb') as fout:
+        try:
+            lines = fout.readlines()
+        except:
+            lines = ''
+
+        if lines is not None:
+            lista = [x.strip() for i, x in enumerate(lines) if i > 0]
+        else:
+            lista = []
+
     try:
         return render_template("index.html", list_todo = lista, len_todo = len(lista))
     except Exception as e: