andis/intern2020
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

pep8

Browse Source
This commit is contained in:
www
2020-12-27 19:12:27 +00:00
parent 35081dee68
commit 8735848ab7
1 changed files with 160 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files

110
bot1/chats.py
View File

@@ -1,11 +1,15 @@
import yaml
from . import CHATFILE, TARGET, URL_HOSTNAME
from pytgbot.bot.sync import SyncBot
from pytgbot.api_types.sendable.inline import InlineQueryResultArticle, InputTextMessageContent
from pytgbot.api_types.sendable.inline import (
InlineQueryResultArticle,
InputTextMessageContent,
)
from pytgbot.api_types.receivable.inline import InlineQuery
from pytgbot.api_types.receivable.updates import Update
from .states import CreatePostWorkflow
from urllib.parse import urljoin
# from . import solr,reindex
from solrfet2020 import SolrFet2020
from . import fet, users
@@ -15,42 +19,50 @@ from key import API_KEY
import requests
from .workflows import AuthWorkflow
import logging
logger = logging.getLogger()
from misc import SaveFileMapping, SaveFileObject
from lxml.html.clean import clean_html, Cleaner
solr = SolrFet2020()
def TelegramIntegrationException(Exception):
pass
def get_chat_id(update):
if update.message and update.message.chat:
return update.message.chat.id
if update.callback_query:
return update.callback_query.message.chat.id
def get_message_id(update):
if update.message:
return update.message.message_id
def get_from_id(update):
if update.message and update.message.to_array():
return str(update.message.to_array()["from"]["id"])
if update.callback_query and update.callback_query.to_array():
return str(update.callback_query.to_array()["from"]["id"])
def get_from(update):
if update.message and update.message.to_array():
return (update.message.to_array()["from"])
return update.message.to_array()["from"]
if update.callback_query and update.callback_query.to_array():
return (update.callback_query.to_array()["from"])
return update.callback_query.to_array()["from"]
def download_file(url):
local_filename = url.split('/')[-1]
local_filename = url.split("/")[-1]
# NOTE the stream=True parameter below
with requests.get(url, stream=True) as r:
r.raise_for_status()
with open(local_filename, 'wb') as f:
with open(local_filename, "wb") as f:
for chunk in r.iter_content(chunk_size=8192):
# If you have chunk encoded response uncomment if
# and set chunk_size parameter to None.
@@ -61,6 +73,7 @@ def download_file(url):
class BaseChat(SaveFileObject):
"Basis für einen Telegram Chat"
def __init__(self, id, bot):
self.id = id
# self.chat_id=chat_id
@@ -70,14 +83,16 @@ class BaseChat(SaveFileObject):
self.debug = False
self.type = "group"
self.mode = ""
def to_dict(self):
return {
"id": self.id,
"workflows": self.workflows,
"debug": self.debug,
"type": self.type,
"mode": self.mode
"mode": self.mode,
}
@classmethod
def from_dict(self, bot, d):
c = Chat(d["id"], bot)
@@ -88,10 +103,18 @@ class BaseChat(SaveFileObject):
def echo(self, update):
self.reply_msg(yaml.dump(update._raw) + "\nfrom" + str(get_from_id(update)))
def send_msg(self, text, **kwargs):
self.bot.send_message(chat_id=self.id, text=text, **kwargs)
def reply_msg(self, text, **kwargs):
self.bot.send_message(chat_id=self.id, text=text, reply_to_message_id=self.last_message_id,**kwargs)
self.bot.send_message(
chat_id=self.id,
text=text,
reply_to_message_id=self.last_message_id,
**kwargs
)
def asdict(self):
return self.to_dict()
@@ -101,6 +124,7 @@ class BaseChat(SaveFileObject):
def log_update(self, update):
with open("chat_logs/%s.yaml" % self.id, "a+") as f:
f.write(yaml.safe_dump([update.to_array()]))
def process_update(self, update):
self.log_update(update)
if update.inline_query:
@@ -126,14 +150,19 @@ class BaseChat(SaveFileObject):
self.bot.edit_message_text(
chat_id=update.callback_query.message.chat.id,
message_id=update.callback_query.message.message_id,
text=update.callback_query.message.text+": "+update.callback_query.data)
text=update.callback_query.message.text
+ ": "
+ update.callback_query.data,
)
if update.message and update.message.photo:
ff=max(update.message.photo, key=lambda x: x.file_size) # Foto mit bester Auflösung
ff = max(
update.message.photo, key=lambda x: x.file_size
) # Foto mit bester Auflösung
fff = self.bot.get_file(ff.file_id)
url = fff.get_download_url(API_KEY)
r = requests.get(url)
fn = url.split('/')[-1]
fn = url.split("/")[-1]
media = (fn, BytesIO(r.content), r.headers["content-type"])
if w:
w.process_message(update.message.text, media=media)
@@ -144,18 +173,22 @@ class BaseChat(SaveFileObject):
class Chat(BaseChat):
def inline(self, query):
# r=solr.search("text_txt: %s" % ,sort="date_dt desc").docs
links, hits = solr.search(query.query)
r = links[0:6]
res=[ InlineQueryResultArticle(
res = [
InlineQueryResultArticle(
id=d["url"],
title=d["title"],
url=urljoin(TARGET, d["url"]),
thumb_url=urljoin(TARGET, d["image"]),
input_message_content = InputTextMessageContent(message_text=urljoin(TARGET,d["url"]))
) for d in r]
input_message_content=InputTextMessageContent(
message_text=urljoin(TARGET, d["url"])
),
)
for d in r
]
# links=[p["title"]+": "+"(%s) "% p["public_date"] + urljoin(TARGET,p["url"]) for p in r]
self.bot.answer_inline_query(inline_query_id=query.id, results=res)
return
@@ -182,7 +215,9 @@ class Chat(BaseChat):
elif cmd == "/debug":
if not u.fet_user:
self.reply_msg("bitte vorher /auth ausführen wenn du ein FET Mitglied bist")
self.reply_msg(
"bitte vorher /auth ausführen wenn du ein FET Mitglied bist"
)
return True
if self.debug:
self.debug = False
@@ -193,14 +228,18 @@ class Chat(BaseChat):
elif cmd == "/post":
if not u.fet_user:
self.reply_msg("bitte vorher /auth ausführen wenn du ein FET Mitglied bist")
self.reply_msg(
"bitte vorher /auth ausführen wenn du ein FET Mitglied bist"
)
return True
self.workflows[get_from_id(update)] = CreatePostWorkflow(chat=self)
return True
elif cmd == "/reindex":
if not u.fet_user:
self.reply_msg("bitte vorher /auth ausführen wenn du ein FET Mitglied bist")
self.reply_msg(
"bitte vorher /auth ausführen wenn du ein FET Mitglied bist"
)
return True
self.reply_msg("Das kann ein bissl dauern...")
solr.reindex()
@@ -208,7 +247,9 @@ class Chat(BaseChat):
return True
elif cmd == "/reindextest":
if not u.fet_user:
self.reply_msg("bitte vorher /auth ausführen wenn du ein FET Mitglied bist")
self.reply_msg(
"bitte vorher /auth ausführen wenn du ein FET Mitglied bist"
)
return True
self.reply_msg("Das kann ein bissl dauern...")
solr.reindextest()
@@ -220,34 +261,47 @@ class Chat(BaseChat):
else:
u.token = "afwerve"
u.a = 1
self.reply_msg(urljoin("https://"+URL_HOSTNAME, "/auth/%s/%s" %(u.id, u.token)))
self.reply_msg(
urljoin("https://" + URL_HOSTNAME, "/auth/%s/%s" % (u.id, u.token))
)
self.send_msg(urljoin("https://", URL_HOSTNAME))
users.to_file()
return True
else: return False
else:
return False
return False
def get_command_from_update(update):
"get the command and the text if there is one"
if update.message and update.message.entities and update.message.entities[0].type=="bot_command":
return update.message.text[update.message.entities[0].offset:update.message.entities[0].offset+update.message.entities[0].length], \
update.message.text[update.message.entities[0].offset+update.message.entities[0].length:].strip()
if (
update.message
and update.message.entities
and update.message.entities[0].type == "bot_command"
):
return (
update.message.text[
update.message.entities[0].offset : update.message.entities[0].offset
+ update.message.entities[0].length
],
update.message.text[
update.message.entities[0].offset + update.message.entities[0].length :
].strip(),
)
return None, None
class ChatManager(SaveFileMapping):
def __init__(self, bot):
super().__init__(file=CHATFILE)
if not type(bot) is SyncBot:
raise TelegramIntegrationException("This ChatManager needs a SyncBot from pytgbot as an argument")
raise TelegramIntegrationException(
"This ChatManager needs a SyncBot from pytgbot as an argument"
)
self.bot = bot
self.from_file()
def __getitem__(self, key):
if type(key) is Update:
return self.__getitem__(get_chat_id(key))