def setUp(self):
original_mongo_uri = settings.MONGODB_URI
original_db = original_mongo_uri.split("/")[-1]
self.test_db = "test" + original_db
test_uri = "/".join(
original_mongo_uri.split("/")[0:-1]) + "/" + self.test_db
self.testing_storage = Storage(uri=test_uri)
def __init__(self,
channels,
year,
month=None,
day=None,
report_extra={},
use_storage=False,
force_collect=False):
"""
force_collect : if use_storage is enable, force the collect even if there is already a report for this searched date
"""
super(CollectArticles, self).__init__()
self.use_storage = use_storage
self.channels = [
Channel() for Channel in
brokenpromises.channels.perform_channels_import(channels)
]
self.date = (year and int(year) or None, month and int(month)
or None, day and int(day) or None)
self.force_collect = force_collect
self.storage = self.use_storage and Storage() or None
self.report_extra = report_extra
def tearDown(self):
Storage().get_connection().drop_database(self.test_db)
def __init__(self):
super(MrClean, self).__init__()
self.storage = Storage()
from flask import Flask, render_template, request, send_file, \
send_from_directory, Response, abort, session, redirect, url_for, make_response, json
from flask.ext.assets import Environment
from flask.ext.login import LoginManager, login_user, login_required, UserMixin, logout_user
from flask.ext.cache import Cache
from rq_dashboard import RQDashboard
from brokenpromises.storage import Storage
from brokenpromises.channels import get_available_channels
from brokenpromises.operations import CollectArticlesAndSendEmail
from brokenpromises.worker import worker
import os
import datetime
STORAGE = Storage()
class CustomFlask(Flask):
jinja_options = Flask.jinja_options.copy()
jinja_options.update(
dict(block_start_string='[%',
block_end_string='%]',
variable_start_string='[[',
variable_end_string=']]',
comment_start_string='[#',
comment_end_string='#]'))
app = CustomFlask(__name__)
app.config.from_envvar("WEBAPP_SETTINGS")
