148 lines
4.5 KiB
Python
148 lines
4.5 KiB
Python
from statemachine import StateMachine, State
|
|
from pytgbot.api_types.sendable.reply_markup import InlineKeyboardButton,InlineKeyboardMarkup
|
|
from . import fet,TARGET
|
|
from urllib.parse import urljoin
|
|
import logging
|
|
logger=logging.getLogger()
|
|
import datetime
|
|
|
|
class TelegramStateMachine(StateMachine):
|
|
def __init__(self,chat=None,*args,**kwargs):
|
|
self.chat = chat
|
|
super().__init__(*args,**kwargs)
|
|
|
|
def send(self,text:str):
|
|
self.chat.send_msg(text)
|
|
|
|
def send_confirm(self, text:str):
|
|
"Ask for a specific confirmation"
|
|
keyboard = [[InlineKeyboardButton("OK", callback_data='ok'),
|
|
InlineKeyboardButton("Repeat", callback_data='retry')],
|
|
[ InlineKeyboardButton("Cancel", callback_data='cancel')] ]
|
|
self.chat.send_msg(text,
|
|
reply_markup=InlineKeyboardMarkup(keyboard))
|
|
|
|
class TelegramWorkflow(TelegramStateMachine):
|
|
def process_message(self, text, is_callback=False):
|
|
pass
|
|
pass
|
|
|
|
|
|
class CreatePostWorkflow(TelegramWorkflow):
|
|
# init --> wait step 0 --> confirm --> wait step 1--> confirm -->wait_photo --> finished
|
|
|
|
init =State('init', initial=True) # Initialize the workflow
|
|
finished = State('finished') # Workflow has ended
|
|
|
|
wait = State('wait',value=0)
|
|
confirm =State('confirm')
|
|
|
|
wait_photo=State('wait_photo')
|
|
|
|
initialize = init.to(wait) # init --> wait
|
|
entered = wait.to(confirm) # wait --> confirm
|
|
next=confirm.to(wait) # confirm --> wait
|
|
retry=confirm.to(wait) # confirm --> wait
|
|
get_photo = confirm.to(wait_photo) # confirm --> wait_photo
|
|
|
|
#finish = wait_photo.to(finished)
|
|
finish = finished.from_(wait_photo, confirm)
|
|
cancel = finished.from_(wait,confirm,wait_photo)
|
|
|
|
steps= \
|
|
[
|
|
{
|
|
"id": "title",
|
|
"question": "Bitte einen Titel für den Post eingeben"
|
|
},
|
|
{
|
|
"id": "body",
|
|
"question": "jetzt bitte den Text"
|
|
}
|
|
]
|
|
|
|
@property
|
|
def current_field(self):
|
|
print(self.steps[self.step])
|
|
return self.steps[self.step]["id"]
|
|
|
|
@property
|
|
def value(self):
|
|
return self.p[self.current_field]
|
|
|
|
def __init__(self,chat=None, debug=False,*args, **kwargs):
|
|
super().__init__(chat, *args, **kwargs)
|
|
self.p=dict()
|
|
self.step=0
|
|
self.debug = debug
|
|
self.initialize()
|
|
|
|
|
|
# processing messages
|
|
def process_message(self, text, is_callback=False,media=None):
|
|
if self.debug: self.send("Workflow Input: %s" % text)
|
|
if self.current_state==self.wait and text:
|
|
self.p[self.current_field]=text
|
|
if self.debug: self.send(str(self.p))
|
|
self.entered()
|
|
elif self.current_state==self.confirm:
|
|
if text=="ok":
|
|
self.confirmed()
|
|
elif text =='cancel':
|
|
self.cancel()
|
|
elif text == 'retry':
|
|
self.retry()
|
|
else:
|
|
self.send("Ich warte noch auf eine Antwort bitte ok/cancel/retry")
|
|
#elif self.current_state == self.wait_photo:
|
|
if media:
|
|
if self.debug:
|
|
self.send("Danke fürs Foto")
|
|
self.p["files"]={"image":media}
|
|
if self.current_state == self.wait_photo:
|
|
self.finish()
|
|
#else:
|
|
# self.retry()
|
|
return ""
|
|
def confirmed(self):
|
|
if self.step >= len(self.steps)-1:
|
|
self.get_photo()
|
|
else:
|
|
self.next()
|
|
|
|
def on_next(self):
|
|
self.step+=1
|
|
|
|
def on_retry(self):
|
|
if self.debug: self.send("retry")
|
|
|
|
def on_cancel(self):
|
|
if self.debug: self.send("Canceled")
|
|
|
|
def on_finish(self):
|
|
#self.step=0
|
|
self.p["legacy_id"]=""
|
|
self.p["post_type"]="N"
|
|
self.p["has_agenda"]="False"
|
|
self.p["public_date"]=datetime.date.today()
|
|
self.send("posting to %s" % TARGET)
|
|
r,d=fet.create(self.p)
|
|
logger.info(r)
|
|
if r ==201:
|
|
self.send("Eingabe fertig:%s" % str(urljoin(TARGET,"posts/"+d["slug"])))
|
|
else:
|
|
self.send("Fehler beim posten -- sorry")
|
|
|
|
def on_enter_confirm(self):
|
|
self.send_confirm("Bitte die Eingabe von %s bestaetigen" % self.value)
|
|
|
|
def on_enter_wait(self):
|
|
self.send(self.steps[self.step]["question"])
|
|
|
|
def on_get_photo(self):
|
|
if not "files" in self.p:
|
|
self.send("Bitte jetzt ein Foto schicken .. ")
|
|
else:
|
|
self.finish()
|
|
|