ticketfrei/ticketfrei3
8
1
Fork 0
Code Issues 11 Pull requests 2 Projects 1 Releases Wiki Activity

[twitter] Fix DMs since_id check

Browse source
This commit is contained in:
Cathy Hu 2020-07-06 11:11:08 +02:00 committed by acipm
parent 3f5f5106e0
commit cc4bf13388
1 changed files with 5 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
kibicara/platforms/twitter/bot.py
View file

@ -25,8 +25,8 @@ class TwitterBot(Censor):
}
self.client = PeonyClient(**self.tokens)
self.polling_interval_sec = 60
self.mentions_since_id = None
self.dms_since_id = None
self.mentions_since_id = self.twitter_model.mentions_since_id
self.dms_since_id = self.twitter_model.dms_since_id
async def run(self):
await gather(self.poll(), self.push())
@ -35,12 +35,10 @@ class TwitterBot(Censor):
while True:
messages = await self._poll_direct_messages()
messages.extend(await self._poll_mentions())
print(messages)
# TODO hold since_ids in database
logger.debug('Polled messages: %s' % str(messages))
await self.twitter_model.update(
dms_since_id=self.dms_since_id, mentions_since_id=self.mentions_since_id
)
# TODO send message to censor
for message in messages:
await self.publish(Message(message))
await sleep(self.polling_interval_sec)
@ -52,10 +50,10 @@ class TwitterBot(Censor):
dms_filtered = []
if dms:
for dm in dms:
if dm.id == self.dms_since_id:
if int(dm.id) == self.dms_since_id:
break
dms_filtered.append(dm)
self.dms_since_id = dms[0].id
self.dms_since_id = int(dms[0].id)
messages = []
for dm in dms_filtered:
filtered_text = await self._filter_text(
@ -65,7 +63,6 @@ class TwitterBot(Censor):
if not filtered_text:
continue
messages.append(filtered_text)
logger.debug(messages)
return messages
async def _poll_mentions(self):
@ -80,7 +77,6 @@ class TwitterBot(Censor):
if not filtered_text:
continue
messages.append(filtered_text)
logger.debug(messages)
return messages
async def _filter_text(self, entities, text):