From bff4dca20dee01c68830faa237e81707ae359036 Mon Sep 17 00:00:00 2001
From: Javinator9889 <javialonso007@hotmail.es>
Date: Tue, 26 Jun 2018 13:58:35 +0200
Subject: [PATCH] DB integration and handlers defined

---
 App/.idea/workspace.xml  | 138 +++++++++++++++++++++++++++++++++------
 App/application.py       |  26 +++++++-
 App/commands/__init__.py |  46 ++++++++++---
 App/utils/__init__.py    |   7 ++
 4 files changed, 184 insertions(+), 33 deletions(-)
 create mode 100644 App/utils/__init__.py

diff --git a/App/.idea/workspace.xml b/App/.idea/workspace.xml
index 336a0a3..ebd1dee 100644
--- a/App/.idea/workspace.xml
+++ b/App/.idea/workspace.xml
@@ -1,7 +1,12 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
   <component name="ChangeListManager">
-    <list default="true" id="ecb85847-acff-4336-b3d7-4f5563aac01d" name="Default" comment="" />
+    <list default="true" id="ecb85847-acff-4336-b3d7-4f5563aac01d" name="Default" comment="">
+      <change afterPath="$PROJECT_DIR$/utils/__init__.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/application.py" beforeDir="false" afterPath="$PROJECT_DIR$/application.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/commands/__init__.py" beforeDir="false" afterPath="$PROJECT_DIR$/commands/__init__.py" afterDir="false" />
+    </list>
     <option name="EXCLUDED_CONVERTED_TO_IGNORED" value="true" />
     <option name="TRACKING_ENABLED" value="true" />
     <option name="SHOW_DIALOG" value="false" />
@@ -80,11 +85,36 @@
   </component>
   <component name="FileEditorManager">
     <leaf SIDE_TABS_SIZE_LIMIT_KEY="300">
-      <file leaf-file-name="application.py" pinned="false" current-in-tab="true">
+      <file leaf-file-name="application.py" pinned="false" current-in-tab="false">
         <entry file="file://$PROJECT_DIR$/application.py">
           <provider selected="true" editor-type-id="text-editor">
-            <state relative-caret-position="51">
-              <caret line="3" column="48" lean-forward="true" selection-start-line="3" selection-start-column="48" selection-end-line="3" selection-end-column="48" />
+            <state relative-caret-position="631">
+              <caret line="83" column="72" lean-forward="true" selection-start-line="83" selection-start-column="72" selection-end-line="83" selection-end-column="72" />
+              <folding>
+                <element signature="e#154#184#0" expanded="true" />
+              </folding>
+            </state>
+          </provider>
+        </entry>
+      </file>
+      <file leaf-file-name="__init__.py" pinned="false" current-in-tab="true">
+        <entry file="file://$PROJECT_DIR$/commands/__init__.py">
+          <provider selected="true" editor-type-id="text-editor">
+            <state relative-caret-position="459">
+              <caret line="27" column="84" lean-forward="true" selection-start-line="27" selection-start-column="84" selection-end-line="27" selection-end-column="84" />
+              <folding>
+                <element signature="e#0#32#0" expanded="true" />
+                <marker date="1530014293071" expanded="true" signature="1397:1406" ph="..." />
+              </folding>
+            </state>
+          </provider>
+        </entry>
+      </file>
+      <file leaf-file-name="__init__.py" pinned="false" current-in-tab="false">
+        <entry file="file://$PROJECT_DIR$/utils/__init__.py">
+          <provider selected="true" editor-type-id="text-editor">
+            <state relative-caret-position="119">
+              <caret line="7" lean-forward="true" selection-start-line="7" selection-end-line="7" />
             </state>
           </provider>
         </entry>
@@ -92,8 +122,8 @@
       <file leaf-file-name="__init__.py" pinned="false" current-in-tab="false">
         <entry file="file://$PROJECT_DIR$/database/__init__.py">
           <provider selected="true" editor-type-id="text-editor">
-            <state relative-caret-position="5763">
-              <caret line="339" column="44" lean-forward="true" selection-start-line="339" selection-start-column="44" selection-end-line="339" selection-end-column="44" />
+            <state relative-caret-position="408">
+              <caret line="44" column="70" lean-forward="true" selection-start-line="44" selection-start-column="70" selection-end-line="44" selection-end-column="70" />
               <folding>
                 <element signature="e#0#11#0" expanded="true" />
               </folding>
@@ -134,10 +164,11 @@
         <option value="$PROJECT_DIR$/out/__init__.py" />
         <option value="$PROJECT_DIR$/requirements.txt" />
         <option value="$PROJECT_DIR$/upgrader/__init__.py" />
-        <option value="$PROJECT_DIR$/commands/__init__.py" />
-        <option value="$PROJECT_DIR$/application.py" />
         <option value="$PROJECT_DIR$/db_script_pycharm.ddl" />
+        <option value="$PROJECT_DIR$/utils/__init__.py" />
+        <option value="$PROJECT_DIR$/application.py" />
         <option value="$PROJECT_DIR$/database/__init__.py" />
+        <option value="$PROJECT_DIR$/commands/__init__.py" />
       </list>
     </option>
   </component>
@@ -170,11 +201,21 @@
               <item name="App" type="b2602c69:ProjectViewProjectNode" />
               <item name="App" type="462c0819:PsiDirectoryNode" />
             </path>
+            <path>
+              <item name="App" type="b2602c69:ProjectViewProjectNode" />
+              <item name="App" type="462c0819:PsiDirectoryNode" />
+              <item name="commands" type="462c0819:PsiDirectoryNode" />
+            </path>
             <path>
               <item name="App" type="b2602c69:ProjectViewProjectNode" />
               <item name="App" type="462c0819:PsiDirectoryNode" />
               <item name="database" type="462c0819:PsiDirectoryNode" />
             </path>
+            <path>
+              <item name="App" type="b2602c69:ProjectViewProjectNode" />
+              <item name="App" type="462c0819:PsiDirectoryNode" />
+              <item name="utils" type="462c0819:PsiDirectoryNode" />
+            </path>
           </expand>
           <select />
         </subPane>
@@ -297,7 +338,7 @@
       <window_info anchor="bottom" id="Run" order="8" weight="0.32965517" />
       <window_info anchor="bottom" id="Terminal" order="4" />
       <window_info anchor="right" id="Remote Host" order="2" />
-      <window_info content_ui="combo" id="Project" order="1" visible="true" weight="0.234375" />
+      <window_info active="true" content_ui="combo" id="Project" order="1" visible="true" weight="0.234375" />
       <window_info anchor="bottom" id="Docker" order="0" show_stripe_button="false" />
       <window_info anchor="right" id="Database" order="0" />
       <window_info anchor="right" id="SciView" order="1" />
@@ -334,10 +375,51 @@
     </breakpoint-manager>
   </component>
   <component name="editorHistoryManager">
+    <entry file="file://$PROJECT_DIR$/application.py">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="1020">
+          <caret line="63" column="63" selection-start-line="63" selection-start-column="63" selection-end-line="63" selection-end-column="63" />
+          <folding>
+            <element signature="e#154#184#0" expanded="true" />
+          </folding>
+        </state>
+      </provider>
+    </entry>
+    <entry file="file://$PROJECT_DIR$/database/__init__.py">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="5644">
+          <caret line="339" column="44" lean-forward="true" selection-start-line="339" selection-start-column="44" selection-end-line="339" selection-end-column="44" />
+          <folding>
+            <element signature="e#0#11#0" expanded="true" />
+          </folding>
+        </state>
+      </provider>
+    </entry>
+    <entry file="file://$PROJECT_DIR$/db_script_pycharm.ddl">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="102">
+          <caret line="6" column="70" selection-start-line="6" selection-start-column="70" selection-end-line="6" selection-end-column="70" />
+        </state>
+      </provider>
+    </entry>
+    <entry file="file://$PROJECT_DIR$/commands/__init__.py">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="578">
+          <caret line="35" column="8" selection-start-line="35" selection-start-column="8" selection-end-line="35" selection-end-column="8" />
+          <folding>
+            <element signature="e#0#32#0" expanded="true" />
+            <marker date="1530014293071" expanded="true" signature="1397:1406" ph="..." />
+          </folding>
+        </state>
+      </provider>
+    </entry>
     <entry file="file://$PROJECT_DIR$/application.py">
       <provider selected="true" editor-type-id="text-editor">
         <state relative-caret-position="119">
           <caret line="7" selection-start-line="7" selection-end-line="7" />
+          <folding>
+            <element signature="e#154#184#0" expanded="true" />
+          </folding>
         </state>
       </provider>
     </entry>
@@ -445,13 +527,6 @@
         </state>
       </provider>
     </entry>
-    <entry file="file://$PROJECT_DIR$/commands/__init__.py">
-      <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="425">
-          <caret line="25" column="61" lean-forward="true" selection-start-line="25" selection-start-column="61" selection-end-line="25" selection-end-column="61" />
-        </state>
-      </provider>
-    </entry>
     <entry file="file://$PROJECT_DIR$/app_data.dict" />
     <entry file="file://$USER_HOME$/.PyCharm2018.1/system/python_stubs/-272786505/builtins.py">
       <provider selected="true" editor-type-id="text-editor">
@@ -467,20 +542,41 @@
         </state>
       </provider>
     </entry>
+    <entry file="file://$PROJECT_DIR$/utils/__init__.py">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="119">
+          <caret line="7" lean-forward="true" selection-start-line="7" selection-end-line="7" />
+        </state>
+      </provider>
+    </entry>
+    <entry file="file://$PROJECT_DIR$/application.py">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="631">
+          <caret line="83" column="72" lean-forward="true" selection-start-line="83" selection-start-column="72" selection-end-line="83" selection-end-column="72" />
+          <folding>
+            <element signature="e#154#184#0" expanded="true" />
+          </folding>
+        </state>
+      </provider>
+    </entry>
     <entry file="file://$PROJECT_DIR$/database/__init__.py">
       <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="5763">
-          <caret line="339" column="44" lean-forward="true" selection-start-line="339" selection-start-column="44" selection-end-line="339" selection-end-column="44" />
+        <state relative-caret-position="408">
+          <caret line="44" column="70" lean-forward="true" selection-start-line="44" selection-start-column="70" selection-end-line="44" selection-end-column="70" />
           <folding>
             <element signature="e#0#11#0" expanded="true" />
           </folding>
         </state>
       </provider>
     </entry>
-    <entry file="file://$PROJECT_DIR$/application.py">
+    <entry file="file://$PROJECT_DIR$/commands/__init__.py">
       <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="51">
-          <caret line="3" column="48" lean-forward="true" selection-start-line="3" selection-start-column="48" selection-end-line="3" selection-end-column="48" />
+        <state relative-caret-position="459">
+          <caret line="27" column="84" lean-forward="true" selection-start-line="27" selection-start-column="84" selection-end-line="27" selection-end-column="84" />
+          <folding>
+            <element signature="e#0#32#0" expanded="true" />
+            <marker date="1530014293071" expanded="true" signature="1397:1406" ph="..." />
+          </folding>
         </state>
       </provider>
     </entry>
diff --git a/App/application.py b/App/application.py
index 14ceb0e..df445c3 100644
--- a/App/application.py
+++ b/App/application.py
@@ -5,6 +5,8 @@
 
 from out import cPrint, Colors
 from upgrader import PiPUpgrader
+from database import DatabaseOperationsBase
+from commands import *
 
 
 def main(arguments: Namespace):
@@ -15,7 +17,8 @@ def main(arguments: Namespace):
         import string
 
         from os import path
-        from telegram.ext import Updater, CommandHandler
+        from telegram import MessageEntity
+        from telegram.ext import Updater, CommandHandler, MessageHandler, Filters
     except (ImportError, ModuleNotFoundError) as import_error:
         print("Modules needed not found: " + str(import_error))
         exit(-1)
@@ -58,13 +61,30 @@ def main(arguments: Namespace):
             upgrader.upgradePackages()
             cPrint("Obtaining values...", Colors.GREEN)
             with open("app_data.dict", "rb") as app_data_file:
-                # unpickler = pickle.Unpickler(app_data_file)
                 app_data = pickle.load(app_data_file)
+            cPrint("Starting database system...", Colors.GREEN)
+            db_manager = DatabaseOperationsBase(username=database_user, password=database_password)
+            cPrint("Defining handlers...", Colors.GREEN)
+            start_handler = CommandHandler("start", )
+            help_handler = CommandHandler("help", )
+            develop_handler = CommandHandler("develop", )
+            video_id_handler = MessageHandler(Filters.command, )
+            url_handler = MessageHandler(Filters.text & (Filters.entity(MessageEntity.URL) |
+                                                         Filters.entity(MessageEntity.TEXT_LINK)), )
+            message_handler = MessageHandler(Filters.text, )
+            unknown_handler = MessageHandler(Filters.all, )
             updater = Updater(token=app_data["TOKEN"], workers=50)
             dispatcher = updater.dispatcher
+
             logging.basicConfig(format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
                                 level=logging.DEBUG)
-            updater.start_polling(poll_interval=5, timeout=60)
+            try:
+                updater.start_polling(poll_interval=5, timeout=60)
+            except KeyboardInterrupt:
+                cPrint("Exiting program... Wait while closing threads and pending petitions...", Colors.FAIL)
+                updater.idle()
+                db_manager.finishConnection()
+                exit(0)
 
 
 if __name__ == '__main__':
diff --git a/App/commands/__init__.py b/App/commands/__init__.py
index a68de43..0fc5bc9 100644
--- a/App/commands/__init__.py
+++ b/App/commands/__init__.py
@@ -1,22 +1,37 @@
 from telegram import Bot, Update
 from telegram.ext import run_async
 
+from utils import Constants
+from database import InsertOperations
 
-class StartHandler:
-    def __init__(self, start_messages: dict):
-        self.__messages = start_messages
 
+class Handler:
+    def __init__(self, handler_messages: dict):
+        self.__messages = handler_messages
+
+class StartHandler(Handler):
     @run_async
     def start(self, bot: Bot, update: Update):
-        chat_id = update.message.chat_id
-        user_id = update.message.user_id
-        bot.sendMessage(chat_id, self.__messages["welcome"])
+        import os
+        import pickle
+
+        effective_user = update.effective_user
+        user_id = effective_user.id
+        username = effective_user.username
+        name = effective_user.first_name
+        user_path = Constants.P_USERS_PATH + user_id + '/'
+        db_insert: InsertOperations = InsertOperations()
+        if not os.path.exists(user_path):
+            os.mkdir(user_path)
+            with open(user_path + Constants.P_USERS_FILE, "wb") as user_info:
+                pickle.dump(Constants.D_USER_DICT, user_info)
+            db_insert.registerNewUser(user_id=user_id, username=username, name=name)
 
 
-class HelpHandler:
-    def __init__(self, help_messages: dict):
-        self.__messages = help_messages
+        # bot.sendMessage(chat_id, self.__messages["welcome"])
 
+
+class HelpHandler(Handler):
     @run_async
     def help(self, bot: Bot, update: Update, args: list):
         chat_id = update.message.chat_id
@@ -24,3 +39,16 @@ def help(self, bot: Bot, update: Update, args: list):
             print("no args")
         else:
             bot.sendMessage(chat_id, self.__messages["help"])
+
+
+class DeveloperHandler(Handler):
+    @run_async
+    def develop(self, bot: Bot, update: Update):
+        # to do
+
+
+class VideoIDHandler(Handler):
+    @run_async
+    def video_handler(self, bot: Bot, update: Update):
+        video_id = update.message.text
+
diff --git a/App/utils/__init__.py b/App/utils/__init__.py
new file mode 100644
index 0000000..55f0bad
--- /dev/null
+++ b/App/utils/__init__.py
@@ -0,0 +1,7 @@
+class Constants:
+    P_USERS_PATH = "user_data/"
+    P_USERS_FILE = "user_info.json"
+
+    D_USER_DICT = {"state": 0,
+                   "is_downloading_video": False,
+                   "pending_videos": 0}