def run(self, curr_day = None, curr_hour = None):
"""Process the list of extensions."""
# Allow these to be passed in for testing
if curr_day is None:
curr_day = int(datetime.datetime.now().strftime("%w"))
if curr_hour is None:
curr_hour = int(datetime.datetime.now().strftime("%H"))
LOGGER.info ("Processing extensions for day segment %d and hour segment %d", curr_day, curr_hour)
my_query = ''.join([
"[[Category:Extension]]",
"[[Has day segment::%d]]" % curr_day,
"[[Has hour segment::%d]]" % curr_hour,
"|sort=Creation date",
"|limit=1000"])
LOGGER.debug ("Query: %s" % my_query)
extensions = self.bumble_bee.call({
'action': 'ask',
'query': my_query
})
i = 0
for extension in extensions['query']['results'].items():
i += 1
LOGGER.info(extension)
LOGGER.info("Processing extension %s", extension[1]['fulltext'])
# Now call tasks to operate on extensions
MediawikiTasks.delay(extension)
return i
def run(self, curr_day=None, curr_hour=None):
"""Process the list of extensions."""
# Allow these to be passed in for testing
if curr_day is None:
curr_day = int(datetime.datetime.now().strftime("%w"))
if curr_hour is None:
curr_hour = int(datetime.datetime.now().strftime("%H"))
LOGGER.info(
"Processing extensions for day segment %d and hour segment %d",
curr_day, curr_hour)
my_query = ''.join([
"[[Category:Extension]]",
"[[Has day segment::%d]]" % curr_day,
"[[Has hour segment::%d]]" % curr_hour, "|sort=Creation date",
"|limit=1000"
])
LOGGER.debug("Query: %s" % my_query)
extensions = self.bumble_bee.call({'action': 'ask', 'query': my_query})
i = 0
for extension in extensions['query']['results'].items():
i += 1
LOGGER.info(extension)
LOGGER.info("Processing extension %s", extension[1]['fulltext'])
# Now call tasks to operate on extensions
MediawikiTasks.delay(extension)
return i
def test_get_mwpagetitle(self):
"""Extension:ConfirmEdit returns Extension:ConfirmEdit"""
task = MediawikiTasks()
assert task.get_mwpagetitle(
"Extension:ConfirmEdit") == "Extension:ConfirmEdit"
assert task.get_mwpagetitle(
"Extension:Confirm User Accounts") == "Extension:ConfirmAccount"
def test_mediawiki_tasks(self):
"""Run the extensions for now."""
task = MediawikiTasks()
task.run("Extension:ConfirmEdit")
def test_get_rating(self):
"""Extension:ConfirmEdit has ratings on it"""
task = MediawikiTasks()
assert task.get_mwpagetitle("Extension:ConfirmEdit") > 0
def test_get_mwpagetitle(self):
"""Extension:ConfirmEdit returns Extension:ConfirmEdit"""
task = MediawikiTasks()
assert task.get_mwpagetitle("Extension:ConfirmEdit") == "Extension:ConfirmEdit"
assert task.get_mwpagetitle("Extension:Confirm User Accounts") == "Extension:ConfirmAccount"
def test_mediawiki_tasks(self):
"""Run the extensions for now."""
task = MediawikiTasks()
task.run("Extension:ConfirmEdit")
def test_get_rating(self):
"""Extension:ConfirmEdit has ratings on it"""
task = MediawikiTasks()
assert task.get_mwpagetitle("Extension:ConfirmEdit") > 0
