added logging and discord message interface

This commit is contained in:
simplypower-bbj
2022-12-17 00:53:33 +01:00
parent 7dfea2891c
commit 9a07ad1dc7
+28
View File
@@ -0,0 +1,28 @@
#!/Users/brian/Projects/twitter_scraper/venv/bin/python
from discord_logging.handler import DiscordHandler
from dotenv import load_dotenv
import logging
import os
def setup_logging():
# setup stream handler
fmt = '[%(asctime)s] %(levelname)s:%(filename)s:%(funcName)s:%(message)s'
datefmt = '%Y-%m-%d %H:%M:%S'
logger_level = os.getenv('LOGGER_LEVEL')
level = getattr(logging, logger_level.upper())
logging.basicConfig(format=fmt, datefmt=datefmt, level=level)
logger = logging.getLogger()
# add discord handler
discord_handler = DiscordHandler(
service_name = 'data_collection',
webhook_url = os.getenv('DISCORD_URL'),
)
discord_handler.setFormatter(logging.Formatter('%(message)s'))
discord_handler.setLevel(logging.WARNING)
logger.addHandler(discord_handler)
logging.debug('finished')
if __name__ == '__main__':
load_dotenv()
setup_logging()
logging.warning('test')