Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| d76e254eeb | |||
| 52a34d0ecc |
22
main.py
22
main.py
@@ -1,9 +1,9 @@
|
|||||||
import os
|
import os
|
||||||
import random
|
import random
|
||||||
from dotenv import load_dotenv
|
from dotenv import load_dotenv
|
||||||
from telethon.sync import TelegramClient
|
|
||||||
import markovify
|
import markovify
|
||||||
from argparse import ArgumentParser
|
from argparse import ArgumentParser
|
||||||
|
import requests
|
||||||
|
|
||||||
parser = ArgumentParser()
|
parser = ArgumentParser()
|
||||||
parser.add_argument("-d", "--dry", action='store_true', help="Run without sending a message")
|
parser.add_argument("-d", "--dry", action='store_true', help="Run without sending a message")
|
||||||
@@ -12,16 +12,14 @@ args = parser.parse_args()
|
|||||||
|
|
||||||
load_dotenv()
|
load_dotenv()
|
||||||
|
|
||||||
api_id = os.getenv('API_ID')
|
access_token = os.getenv('ACCESS_TOKEN')
|
||||||
api_hash = os.getenv('API_HASH')
|
|
||||||
entity = os.getenv('ENTITY')
|
|
||||||
dry_run = args.dry
|
dry_run = args.dry
|
||||||
filename = args.model
|
filename = args.model
|
||||||
|
|
||||||
def create_message(model):
|
def create_message(model):
|
||||||
message = ''
|
message = ''
|
||||||
|
|
||||||
for i in range(5):
|
for i in range(4):
|
||||||
try:
|
try:
|
||||||
message = message + model.make_sentence()
|
message = message + model.make_sentence()
|
||||||
except TypeError:
|
except TypeError:
|
||||||
@@ -44,5 +42,15 @@ model = markovify.Text.from_json(model_json)
|
|||||||
if (dry_run):
|
if (dry_run):
|
||||||
print(create_message(model))
|
print(create_message(model))
|
||||||
else:
|
else:
|
||||||
with TelegramClient('bot', api_id, api_hash) as client:
|
data = {
|
||||||
client.send_message(entity=entity, message=create_message(model))
|
'chat_id': 'channel_608',
|
||||||
|
'text': create_message(model),
|
||||||
|
}
|
||||||
|
response = requests.post(
|
||||||
|
'https://arround-msg.gora.studio/v3/message/message',
|
||||||
|
json = data,
|
||||||
|
headers = {
|
||||||
|
'x-access-token': access_token
|
||||||
|
}
|
||||||
|
)
|
||||||
|
print(response.content)
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
markovify
|
markovify
|
||||||
telethon
|
requests
|
||||||
python-dotenv
|
python-dotenv
|
||||||
argparse
|
argparse
|
||||||
|
|||||||
Reference in New Issue
Block a user