decentral1se/xbotlib
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Implement storage with Redis

Browse Source 
Closes https://git.autonomic.zone/decentral1se/xbotlib/issues/21.
This commit is contained in:
Luke Murphy 2021-01-15 21:14:04 +01:00
parent 66e6bde2f4
commit af292c45bf
No known key found for this signature in database
GPG Key ID: 5E2EF5A63E3718CC
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
xbotlib.py
View File

@ -9,9 +9,11 @@ from logging import DEBUG, INFO, basicConfig, getLogger
from os import environ from os import environ
from os.path import exists from os.path import exists
from pathlib import Path from pathlib import Path
from random import choice
from sys import exit, stdout from sys import exit, stdout
from humanize import naturaldelta from humanize import naturaldelta
from redis import Redis
from slixmpp import ClientXMPP from slixmpp import ClientXMPP
@ -82,6 +84,7 @@ class Bot(ClientXMPP):
self.init_bot() self.init_bot()
self.register_xmpp_event_handlers() self.register_xmpp_event_handlers()
self.register_xmpp_plugins() self.register_xmpp_plugins()
self.init_db()
self.run() self.run()
def parse_arguments(self): def parse_arguments(self):
@ -282,6 +285,17 @@ class Bot(ClientXMPP):
self.register_plugin("xep_0199") # XMPP Ping self.register_plugin("xep_0199") # XMPP Ping
self.register_plugin("xep_0084") # User Avatar self.register_plugin("xep_0084") # User Avatar
def init_db(self):
"""Initialise the Redis key/value store."""
url = environ.get("REDIS_URL", None)
if not url:
self.db = None
self.log.info("No storage discovered")
else:
self.db = Redis.from_url(url, decode_responses=True)
self.log.info("Successfully connected to storage")
def run(self): def run(self):
"""Run the bot.""" """Run the bot."""
self.connect() self.connect()