Fix logging logic in scraper controller

This commit is contained in:
Logan Williams
2022-02-24 18:49:06 +01:00
parent e64d845002
commit d163e6b3d9
2 changed files with 8 additions and 5 deletions

View File

@@ -46,12 +46,12 @@ class ScraperController:
if not handled:
logger.warning(f"No handler found for Channel {channel}")
else:
session = self.session()
session.bulk_save_objects(posts)
session.commit()
session = self.session()
session.bulk_save_objects(posts)
session.commit()
logger.info(f"Added {len(posts)} entries to database")
logger.info(f"Added {len(posts)} entries to database")
def connect_to_db(self, engine):
# create tables

View File

@@ -20,6 +20,9 @@ class TelegramSnscrapeScraper(cisticola.scraper.base.Scraper):
g = scr.get_items()
for post in g:
if (len(posts)) >= 10:
break
if since is not None and post.date.replace(tzinfo=timezone.utc) <= since.date.replace(tzinfo=timezone.utc):
break