Catch none type sentences

2022-04-03 10:55:20 +03:00
parent 7a1132445f
commit 00b1287ef3
3 changed files with 6 additions and 3 deletions
--- a/.gitignore
+++ b/.gitignore
@@ -4,3 +4,4 @@ venv
 .env
 /data/*
 *.session
+*.session-journal
--- a/grab_samples.py
+++ b/grab_samples.py
@@ -8,8 +8,7 @@ api_id = os.getenv('API_ID')
 api_hash = os.getenv('API_HASH')
 chats = [
  'tg1337const',
-  'cyberboec',
-  'Ateobreaking'
+  'cyberboec'
 ]

 def cleanup(msg):
--- a/main.py
+++ b/main.py
@@ -22,7 +22,10 @@ text_model = markovify.Text(text)
 message = ''

 for i in range(5):
-  message = message + text_model.make_sentence()
+  try:
+    message = message + text_model.make_sentence()
+  except:
+    pass

 if (dry_run):
  print(message)