Merge pull request #111 from andresdelfino/add_repository_to_project

Add repository URL and Telegram group URL to project

Author: WinnaZ

migrations/migrate_to_repository_and_group_urls.py

@@ -0,0 +1,24 @@
+# https://docs.peewee-orm.com/en/latest/peewee/playhouse.html#schema-migrations
+
+import peewee
+import playhouse.migrate
+
+import pycamp_bot.models
+
+
+my_db = peewee.SqliteDatabase('pycamp_projects.db')
+migrator = playhouse.migrate.SqliteMigrator(my_db)
+
+
+playhouse.migrate.migrate(
+    migrator.add_column(
+        pycamp_bot.models.Project._meta.table_name, 
+        'repository_url', 
+        pycamp_bot.models.Project.repository_url,
+    ),
+    migrator.add_column(
+        pycamp_bot.models.Project._meta.table_name, 
+        'group_url', 
+        pycamp_bot.models.Project.group_url,
+    ),
+)

src/pycamp_bot/commands/base.py

@@ -2,13 +2,16 @@
 from pycamp_bot.commands.help_msg import get_help
 from pycamp_bot.logger import logger
 
+import os
+
 
 async def msg_to_active_pycamp_chat(bot, text):
-    chat_id = -1001404878013  # Prueba
-    await bot.send_message(
-        chat_id=chat_id,
-        text=text
-        )
+    if 'TEST_CHAT_ID' in os.environ:
+        chat_id = -1001404878013  # Prueba
+        await bot.send_message(
+            chat_id=os.environ['TEST_CHAT_ID'],
+            text=text
+            )
 
 
 async def start(update, context):

src/pycamp_bot/commands/manage_pycamp.py

@@ -152,6 +152,7 @@ async def define_duration(update, context):
         chat_id=update.message.chat_id,
         text=msg
     )
+    return ConversationHandler.END
 
 
 async def cancel(update, context):