Add bot that automatically tweets some things

[ipfire.org.git] / src / backend / base.py

diff --git a/src/backend/base.py b/src/backend/base.py
index d7d6e328984da06dcbccd60fc6e5b977bbefdb73..6c79d8bf44406e4de7d85241b107a6ebabe1ef42 100644 (file)
--- a/src/backend/base.py
+++ b/src/backend/base.py
@@ -1,22 +1,27 @@
 #!/usr/bin/python
 
-import configparser as configparser
+import configparser
 import io
+import tornado.gen
 
 from . import accounts
+from . import blog
 from . import database
 from . import geoip
 from . import fireinfo
 from . import iuse
+from . import memcached
+from . import messages
 from . import mirrors
 from . import netboot
 from . import nopaste
 from . import releases
 from . import settings
 from . import talk
-
-from . import blog
+from . import tweets
+from . import wiki
 from . import zeiterfassung
+from .decorators import *
 
 DEFAULT_CONFIG = io.StringIO("""
 [global]
@@ -40,10 +45,10 @@ class Backend(object):
 
                # Initialize settings first.
                self.settings = settings.Settings(self)
+               self.memcache = memcached.Memcached(self)
 
                # Initialize backend modules.
                self.accounts = accounts.Accounts(self)
-               self.downloads = mirrors.Downloads(self)
                self.geoip = geoip.GeoIP(self)
                self.fireinfo = fireinfo.Fireinfo(self)
                self.iuse = iuse.IUse(self)
@@ -54,6 +59,7 @@ class Backend(object):
                self.talk = talk.Talk(self)
 
                self.blog = blog.Blog(self)
+               self.wiki = wiki.Wiki(self)
                self.zeiterfassung = zeiterfassung.ZeiterfassungClient(self)
 
        def read_config(self, configfile):
@@ -79,3 +85,35 @@ class Backend(object):
                }
 
                self.db = database.Connection(**credentials)
+
+       @tornado.gen.coroutine
+       def run_task(self, task, *args, **kwargs):
+               tasks = {
+                       "check-mirrors"     : self.mirrors.check_all,
+                       "cleanup-messages"  : self.messages.queue.cleanup,
+                       "scan-files"        : self.releases.scan_files,
+                       "send-all-messages" : self.messages.queue.send_all,
+                       "tweet"             : self.tweets.tweet,
+                       "update-blog-feeds" : self.blog.update_feeds,
+               }
+
+               # Get the task from the list of all tasks
+               func = tasks.get(task, None)
+               if not func:
+                       raise ValueError("Unknown task: %s" % task)
+
+               # Run the task
+               r = yield func(*args, **kwargs)
+
+               # If any error code has been returned,
+               # we will end the program
+               if r:
+                       raise SystemExit(r)
+
+       @lazy_property
+       def messages(self):
+               return messages.Messages(self)
+
+       @lazy_property
+       def tweets(self):
+               return tweets.Tweets(self)