Support sending HTML emails

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

diff --git a/src/backend/base.py b/src/backend/base.py
index 9e4ae21b2578626d39758f2362e4e7afb0414b7b..1578a01544f9a44078f0a744b0e85c2b7f9682eb 100644 (file)
--- a/src/backend/base.py
+++ b/src/backend/base.py
@@ -2,23 +2,28 @@
 
 import configparser
 import io
-import tornado.gen
+import tornado.httpclient
 
 from . import accounts
+from . import blog
+from . import campaigns
 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 ratelimit
 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,8 +45,15 @@ class Backend(object):
                # Setup database.
                self.setup_database()
 
+               # Create HTTPClient
+               self.http_client = tornado.httpclient.AsyncHTTPClient(
+                       defaults = {
+                               "User-Agent" : "IPFireWebApp",
+                       }
+               )
                # Initialize settings first.
                self.settings = settings.Settings(self)
+               self.memcache = memcached.Memcached(self)
 
                # Initialize backend modules.
                self.accounts = accounts.Accounts(self)
@@ -82,9 +94,19 @@ class Backend(object):
 
                self.db = database.Connection(**credentials)
 
-       @tornado.gen.coroutine
-       def run_task(self, task, *args, **kwargs):
+       async def run_task(self, task, *args, **kwargs):
                tasks = {
+                       "check-mirrors"     : self.mirrors.check_all,
+                       "check-spam"        : self.accounts.check_spam,
+                       "cleanup"           : self.cleanup,
+                       "launch-campaigns"  : self.campaigns.launch_manually,
+                       "run-campaigns"     : self.campaigns.run,
+                       "scan-files"        : self.releases.scan_files,
+                       "send-message"      : self.messages.send_cli,
+                       "send-all-messages" : self.messages.queue.send_all,
+                       "test-blacklist"    : self.geoip.test_blacklist,
+                       "test-ldap"         : self.accounts.test_ldap,
+                       "tweet"             : self.tweets.tweet,
                        "update-blog-feeds" : self.blog.update_feeds,
                }
 
@@ -94,9 +116,38 @@ class Backend(object):
                        raise ValueError("Unknown task: %s" % task)
 
                # Run the task
-               r = yield func(*args, **kwargs)
+               r = await func(*args, **kwargs)
 
                # If any error code has been returned,
                # we will end the program
                if r:
                        raise SystemExit(r)
+
+       @lazy_property
+       def campaigns(self):
+               return campaigns.Campaigns(self)
+
+       @lazy_property
+       def groups(self):
+               return accounts.Groups(self)
+
+       @lazy_property
+       def messages(self):
+               return messages.Messages(self)
+
+       @lazy_property
+       def ratelimiter(self):
+               return ratelimit.RateLimiter(self)
+
+       @lazy_property
+       def tweets(self):
+               return tweets.Tweets(self)
+
+       async def cleanup(self):
+               # Cleanup message queue
+               with self.db.transaction():
+                       self.messages.queue.cleanup()
+
+               # Cleanup in accounts
+               with self.db.transaction():
+                       self.accounts.cleanup()