adding get_session function and removed the previous __main__ testing…

… code!

.gitignore

@@ -220,5 +220,6 @@ $RECYCLE.BIN/
 *.lnk
 
 *.db
+flomo/data/*
 
 # End of https://www.toptal.com/developers/gitignore/api/windows,python,visualstudiocode

flomo/tracker.py

@@ -44,18 +44,6 @@ def get_sessions(self):
         self.cursor.execute("SELECT * FROM sessions")
         return self.cursor.fetchall()
 
-
-if __name__ == "__main__":
-    tracker = Tracker()
-    tracker.create_table()
-
-    session_id = tracker.create_session("study", "work", datetime.datetime.now())
-
-    import time
-
-    time.sleep(5)
-
-    tracker.end_session(session_id, datetime.datetime.now())
-
-    print(tracker.get_sessions())
-    tracker.conn.close()
+    def get_session(self, session_id: float):
+        self.cursor.execute("SELECT * FROM sessions WHERE id = ?", (session_id,))
+        return self.cursor.fetchone()