From 39b5424bec7cdf800058524f5b80e7ad6ce62078 Mon Sep 17 00:00:00 2001
From: ale <ale@incal.net>
Date: Tue, 28 Jan 2014 20:16:33 +0000
Subject: [PATCH] make the receiver app compatible with wsgi-app-installer
---
{contrib => offlinescan}/ai/receiver.py | 19 +++++++++++--------
1 file changed, 11 insertions(+), 8 deletions(-)
rename {contrib => offlinescan}/ai/receiver.py (68%)
diff --git a/contrib/ai/receiver.py b/offlinescan/ai/receiver.py
similarity index 68%
rename from contrib/ai/receiver.py
rename to offlinescan/ai/receiver.py
index 72d13cf..0aedd31 100644
--- a/contrib/ai/receiver.py
+++ b/offlinescan/ai/receiver.py
@@ -10,7 +10,6 @@ import redis
from flask import Flask, request, abort, g
app = Flask(__name__)
-app.config.from_pyfile('config.py')
def get_redis_connection():
@@ -33,23 +32,27 @@ def receive():
abort(400)
p = g.redis.pipeline(transaction=False)
- per_user = {}
+ #per_user = {}
for user, results in data.iteritems():
key = 'virus:%s' % user
- p.ltrim(key, 0, 0)
+ #p.ltrim(key, 0, 0)
for r in results:
p.lpush(key, r)
- per_user[user] = len(results)
+ #per_user[user] = len(results)
+
+ # Save some incremental stats with per-user infection counts.
+ #new_ptr = 'virus_stats:%f' % time.time()
+ #for user, count in per_user.iteritems():
+ # p.hset(new_ptr, user, count)
+ #p.set('virus_stats:head', new_ptr)
- new_ptr = 'virus_stats:%f' % time.time()
- for user, count in per_user.iteritems():
- p.hset(new_ptr, user, count)
- p.set('virus_stats:head', new_ptr)
p.execute()
return ['ok']
def create_app():
+ # The env variable name is compatible with wsgi-app-installer.
+ app.config.from_envvar('APP_CONFIG', silent=True)
return app
--
GitLab