From 1f7ab0c2cbe8a8d7ecf21c35faee6d00d0bcc174 Mon Sep 17 00:00:00 2001
From: Anatoly <akopyl@radner.ru>
Date: Sat, 2 Apr 2022 23:17:23 +0300
Subject: [PATCH] Added a dry run option

---
 main.py          | 18 +++++++++++++-----
 requirements.txt |  1 +
 2 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/main.py b/main.py
index 107c494..ded48ed 100644
--- a/main.py
+++ b/main.py
@@ -2,15 +2,21 @@ import os
 from dotenv import load_dotenv
 from telethon.sync import TelegramClient
 import markovify
+from argparse import ArgumentParser
+
+parser = ArgumentParser()
+parser.add_argument("-d", "--dry", action='store_true', help="Run without sending a message")
+args = parser.parse_args()
 
 load_dotenv()
 
 api_id = os.getenv('API_ID')
 api_hash = os.getenv('API_HASH')
 entity = os.getenv('ENTITY')
+dry_run = args.dry
 
-f = open("data/data.txt", "r")
-text = f.read()
+with open("data/data.txt", "r") as f:
+  text = f.read()
 
 text_model = markovify.Text(text)
 message = ''
@@ -18,6 +24,8 @@ message = ''
 for i in range(5):
   message = message + text_model.make_sentence()
 
-with TelegramClient('bot', api_id, api_hash) as client:
-  client.send_message(entity=entity, message=message)
-  
\ No newline at end of file
+if (dry_run):
+  print(message)
+else:
+  with TelegramClient('bot', api_id, api_hash) as client:
+    client.send_message(entity=entity, message=message)
diff --git a/requirements.txt b/requirements.txt
index 45f06a2..6273d89 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
 markovify
 telethon
 python-dotenv
+argparse