Additional IRC commands and bot tasks for EarwigBot https://en.wikipedia.org/wiki/User:EarwigBot
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

afc_statistics.py 30 KiB

11 年之前
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733
  1. # -*- coding: utf-8 -*-
  2. #
  3. # Copyright (C) 2009-2013 Ben Kurtovic <ben.kurtovic@verizon.net>
  4. #
  5. # Permission is hereby granted, free of charge, to any person obtaining a copy
  6. # of this software and associated documentation files (the "Software"), to deal
  7. # in the Software without restriction, including without limitation the rights
  8. # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  9. # copies of the Software, and to permit persons to whom the Software is
  10. # furnished to do so, subject to the following conditions:
  11. #
  12. # The above copyright notice and this permission notice shall be included in
  13. # all copies or substantial portions of the Software.
  14. #
  15. # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16. # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17. # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18. # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19. # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20. # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  21. # SOFTWARE.
  22. from datetime import datetime
  23. import re
  24. from os.path import expanduser
  25. from threading import Lock
  26. from time import sleep
  27. import oursql
  28. from earwigbot import exceptions
  29. from earwigbot import wiki
  30. from earwigbot.tasks import Task
  31. class AFCStatistics(Task):
  32. """A task to generate statistics for WikiProject Articles for Creation.
  33. Statistics are stored in a MySQL database ("u_earwig_afc_statistics")
  34. accessed with oursql. Statistics are synchronied with the live database
  35. every four minutes and saved once an hour, on the hour, to self.pagename.
  36. In the live bot, this is "Template:AFC statistics".
  37. """
  38. name = "afc_statistics"
  39. number = 2
  40. # Chart status number constants:
  41. CHART_NONE = 0
  42. CHART_PEND = 1
  43. CHART_DRAFT = 2
  44. CHART_REVIEW = 3
  45. CHART_ACCEPT = 4
  46. CHART_DECLINE = 5
  47. CHART_MISPLACE = 6
  48. def setup(self):
  49. self.cfg = cfg = self.config.tasks.get(self.name, {})
  50. # Set some wiki-related attributes:
  51. self.pagename = cfg.get("page", "Template:AFC statistics")
  52. self.pending_cat = cfg.get("pending", "Pending AfC submissions")
  53. self.ignore_list = cfg.get("ignoreList", [])
  54. default_summary = "Updating statistics for [[WP:WPAFC|WikiProject Articles for creation]]."
  55. self.summary = self.make_summary(cfg.get("summary", default_summary))
  56. # Templates used in chart generation:
  57. templates = cfg.get("templates", {})
  58. self.tl_header = templates.get("header", "AFC statistics/header")
  59. self.tl_row = templates.get("row", "#invoke:AfC|row")
  60. self.tl_footer = templates.get("footer", "AFC statistics/footer")
  61. # Connection data for our SQL database:
  62. kwargs = cfg.get("sql", {})
  63. kwargs["read_default_file"] = expanduser("~/.my.cnf")
  64. self.conn_data = kwargs
  65. self.db_access_lock = Lock()
  66. def run(self, **kwargs):
  67. """Entry point for a task event.
  68. Depending on the kwargs passed, we will either synchronize our local
  69. statistics database with the site (self.sync()) or save it to the wiki
  70. (self.save()). We will additionally create an SQL connection with our
  71. local database.
  72. """
  73. action = kwargs.get("action")
  74. if not self.db_access_lock.acquire(False): # Non-blocking
  75. if action == "sync":
  76. self.logger.info("A sync is already ongoing; aborting")
  77. return
  78. self.logger.info("Waiting for database access lock")
  79. self.db_access_lock.acquire()
  80. try:
  81. self.site = self.bot.wiki.get_site()
  82. self.conn = oursql.connect(**self.conn_data)
  83. self.revision_cache = {}
  84. try:
  85. if action == "save":
  86. self.save(kwargs)
  87. elif action == "sync":
  88. self.sync(kwargs)
  89. elif action == "update":
  90. self.update(kwargs)
  91. finally:
  92. self.conn.close()
  93. finally:
  94. self.db_access_lock.release()
  95. def save(self, kwargs):
  96. """Save our local statistics to the wiki.
  97. After checking for emergency shutoff, the statistics chart is compiled,
  98. and then saved to self.pagename using self.summary iff it has changed
  99. since last save.
  100. """
  101. self.logger.info("Saving chart")
  102. if kwargs.get("fromIRC"):
  103. summary = self.summary + " (!earwigbot)"
  104. else:
  105. if self.shutoff_enabled():
  106. return
  107. summary = self.summary
  108. statistics = self.compile_charts()
  109. page = self.site.get_page(self.pagename)
  110. text = page.get()
  111. newtext = re.sub(u"<!-- stat begin -->(.*?)<!-- stat end -->",
  112. "<!-- stat begin -->\n" + statistics + "\n<!-- stat end -->",
  113. text, flags=re.DOTALL)
  114. if newtext == text:
  115. self.logger.info("Chart unchanged; not saving")
  116. return # Don't edit the page if we're not adding anything
  117. newtext = re.sub("<!-- sig begin -->(.*?)<!-- sig end -->",
  118. "<!-- sig begin -->~~~ at ~~~~~<!-- sig end -->",
  119. newtext)
  120. page.edit(newtext, summary, minor=True, bot=True)
  121. self.logger.info(u"Chart saved to [[{0}]]".format(page.title))
  122. def compile_charts(self):
  123. """Compile and return all statistics information from our local db."""
  124. stats = ""
  125. with self.conn.cursor() as cursor:
  126. cursor.execute("SELECT * FROM chart")
  127. for chart in cursor:
  128. stats += self.compile_chart(chart) + "\n"
  129. return stats[:-1] # Drop the last newline
  130. def compile_chart(self, chart_info):
  131. """Compile and return a single statistics chart."""
  132. chart_id, chart_title, special_title = chart_info
  133. chart = self.tl_header + "|" + chart_title
  134. if special_title:
  135. chart += "|" + special_title
  136. chart = "{{" + chart + "}}"
  137. query = "SELECT * FROM page JOIN row ON page_id = row_id WHERE row_chart = ?"
  138. with self.conn.cursor(oursql.DictCursor) as cursor:
  139. cursor.execute(query, (chart_id,))
  140. for page in cursor:
  141. chart += "\n" + self.compile_chart_row(page)
  142. chart += "\n{{" + self.tl_footer + "}}"
  143. return chart
  144. def compile_chart_row(self, page):
  145. """Compile and return a single chart row.
  146. 'page' is a dict of page information, taken as a row from the page
  147. table, where keys are column names and values are their cell contents.
  148. """
  149. row = u"{0}|s={page_status}|t={page_title}|z={page_size}|"
  150. if page["page_special_oldid"]:
  151. row += "sr={page_special_user}|sd={page_special_time}|si={page_special_oldid}|"
  152. row += "mr={page_modify_user}|md={page_modify_time}|mi={page_modify_oldid}"
  153. page["page_special_time"] = self.format_time(page["page_special_time"])
  154. page["page_modify_time"] = self.format_time(page["page_modify_time"])
  155. if page["page_notes"]:
  156. row += "|n=1{page_notes}"
  157. return "{{" + row.format(self.tl_row, **page) + "}}"
  158. def format_time(self, dt):
  159. """Format a datetime into the standard MediaWiki timestamp format."""
  160. return dt.strftime("%H:%M, %d %b %Y")
  161. def sync(self, kwargs):
  162. """Synchronize our local statistics database with the site.
  163. Syncing involves, in order, updating tracked submissions that have
  164. been changed since last sync (self.update_tracked()), adding pending
  165. submissions that are not tracked (self.add_untracked()), and removing
  166. old submissions from the database (self.delete_old()).
  167. The sync will be canceled if SQL replication lag is greater than 600
  168. seconds, because this will lead to potential problems and outdated
  169. data, not to mention putting demand on an already overloaded server.
  170. Giving sync the kwarg "ignore_replag" will go around this restriction.
  171. """
  172. self.logger.info("Starting sync")
  173. replag = self.site.get_replag()
  174. self.logger.debug("Server replag is {0}".format(replag))
  175. if replag > 600 and not kwargs.get("ignore_replag"):
  176. msg = "Sync canceled as replag ({0} secs) is greater than ten minutes"
  177. self.logger.warn(msg.format(replag))
  178. return
  179. with self.conn.cursor() as cursor:
  180. self.update_tracked(cursor)
  181. self.add_untracked(cursor)
  182. self.delete_old(cursor)
  183. self.logger.info("Sync completed")
  184. def update_tracked(self, cursor):
  185. """Update tracked submissions that have been changed since last sync.
  186. This is done by iterating through every page in our database and
  187. comparing our stored latest revision ID with the actual latest revision
  188. ID from an SQL query. If they differ, we will update our information
  189. about the page (self.update_page()).
  190. If the page does not exist, we will remove it from our database with
  191. self.untrack_page().
  192. """
  193. self.logger.debug("Updating tracked submissions")
  194. query = """SELECT s.page_id, s.page_title, s.page_modify_oldid,
  195. r.page_latest, r.page_title, r.page_namespace
  196. FROM page AS s
  197. LEFT JOIN {0}_p.page AS r ON s.page_id = r.page_id
  198. WHERE s.page_modify_oldid != r.page_latest"""
  199. cursor.execute(query.format(self.site.name))
  200. for pageid, title, oldid, real_oldid, real_title, real_ns in cursor:
  201. if not real_oldid:
  202. self.untrack_page(cursor, pageid)
  203. continue
  204. msg = u"Updating page [[{0}]] (id: {1}) @ {2}"
  205. self.logger.debug(msg.format(title, pageid, oldid))
  206. msg = u" {0}: oldid: {1} -> {2}"
  207. self.logger.debug(msg.format(pageid, oldid, real_oldid))
  208. real_title = real_title.decode("utf8").replace("_", " ")
  209. ns = self.site.namespace_id_to_name(real_ns)
  210. if ns:
  211. real_title = u":".join((ns, real_title))
  212. try:
  213. self.update_page(cursor, pageid, real_title)
  214. except Exception:
  215. e = u"Error updating page [[{0}]] (id: {1})"
  216. self.logger.exception(e.format(real_title, pageid))
  217. def add_untracked(self, cursor):
  218. """Add pending submissions that are not yet tracked.
  219. This is done by compiling a list of all currently tracked submissions
  220. and iterating through all members of self.pending_cat via SQL. If a
  221. page in the pending category is not tracked and is not in
  222. self.ignore_list, we will track it with self.track_page().
  223. """
  224. self.logger.debug("Adding untracked pending submissions")
  225. query = """SELECT r.page_id, r.page_title, r.page_namespace
  226. FROM {0}_p.page AS r
  227. INNER JOIN {0}_p.categorylinks AS c ON r.page_id = c.cl_from
  228. LEFT JOIN page AS s ON r.page_id = s.page_id
  229. WHERE s.page_id IS NULL AND c.cl_to = ?"""
  230. cursor.execute(query.format(self.site.name),
  231. (self.pending_cat.replace(" ", "_"),))
  232. for pageid, title, ns in cursor:
  233. title = title.decode("utf8").replace("_", " ")
  234. ns = self.site.namespace_id_to_name(ns)
  235. if ns:
  236. title = u":".join((ns, title))
  237. if title in self.ignore_list:
  238. continue
  239. msg = u"Tracking page [[{0}]] (id: {1})".format(title, pageid)
  240. self.logger.debug(msg)
  241. try:
  242. self.track_page(cursor, pageid, title)
  243. except Exception:
  244. e = u"Error tracking page [[{0}]] (id: {1})"
  245. self.logger.exception(e.format(title, pageid))
  246. def delete_old(self, cursor):
  247. """Remove old submissions from the database.
  248. "Old" is defined as a submission that has been declined or accepted
  249. more than 36 hours ago. Pending submissions cannot be "old".
  250. """
  251. self.logger.debug("Removing old submissions from chart")
  252. query = """DELETE FROM page, row USING page JOIN row
  253. ON page_id = row_id WHERE row_chart IN (?, ?)
  254. AND ADDTIME(page_special_time, '36:00:00') < NOW()"""
  255. cursor.execute(query, (self.CHART_ACCEPT, self.CHART_DECLINE))
  256. def update(self, kwargs):
  257. """Update a page by name, regardless of whether anything has changed.
  258. Mainly intended as a command to be used via IRC, e.g.:
  259. !tasks start afc_statistics action=update page=Foobar
  260. """
  261. title = kwargs.get("page")
  262. if not title:
  263. return
  264. title = title.replace("_", " ").decode("utf8")
  265. query = "SELECT page_id, page_modify_oldid FROM page WHERE page_title = ?"
  266. with self.conn.cursor() as cursor:
  267. cursor.execute(query, (title,))
  268. try:
  269. pageid, oldid = cursor.fetchall()[0]
  270. except IndexError:
  271. msg = u"Page [[{0}]] not found in database".format(title)
  272. self.logger.error(msg)
  273. msg = u"Updating page [[{0}]] (id: {1}) @ {2}"
  274. self.logger.info(msg.format(title, pageid, oldid))
  275. self.update_page(cursor, pageid, title)
  276. def untrack_page(self, cursor, pageid):
  277. """Remove a page, given by ID, from our database."""
  278. self.logger.debug("Untracking page (id: {0})".format(pageid))
  279. query = """DELETE FROM page, row USING page JOIN row
  280. ON page_id = row_id WHERE page_id = ?"""
  281. cursor.execute(query, (pageid,))
  282. def track_page(self, cursor, pageid, title):
  283. """Update hook for when page is not in our database.
  284. A variety of SQL queries are used to gather information about the page,
  285. which is then saved to our database.
  286. """
  287. content = self.get_content(title)
  288. if content is None:
  289. msg = u"Could not get page content for [[{0}]]".format(title)
  290. self.logger.error(msg)
  291. return
  292. namespace = self.site.get_page(title).namespace
  293. status, chart = self.get_status_and_chart(content, namespace)
  294. if chart == self.CHART_NONE:
  295. msg = u"Could not find a status for [[{0}]]".format(title)
  296. self.logger.warn(msg)
  297. return
  298. size = self.get_size(content)
  299. m_user, m_time, m_id = self.get_modify(pageid)
  300. s_user, s_time, s_id = self.get_special(pageid, chart)
  301. notes = self.get_notes(chart, content, m_time, s_user)
  302. query1 = "INSERT INTO row VALUES (?, ?)"
  303. query2 = "INSERT INTO page VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)"
  304. cursor.execute(query1, (pageid, chart))
  305. cursor.execute(query2, (pageid, status, title, size, notes, m_user,
  306. m_time, m_id, s_user, s_time, s_id))
  307. def update_page(self, cursor, pageid, title):
  308. """Update hook for when page is already in our database.
  309. A variety of SQL queries are used to gather information about the page,
  310. which is compared against our stored information. Differing information
  311. is then updated.
  312. """
  313. content = self.get_content(title)
  314. if content is None:
  315. msg = u"Could not get page content for [[{0}]]".format(title)
  316. self.logger.error(msg)
  317. return
  318. namespace = self.site.get_page(title).namespace
  319. status, chart = self.get_status_and_chart(content, namespace)
  320. if chart == self.CHART_NONE:
  321. self.untrack_page(cursor, pageid)
  322. return
  323. query = "SELECT * FROM page JOIN row ON page_id = row_id WHERE page_id = ?"
  324. with self.conn.cursor(oursql.DictCursor) as dict_cursor:
  325. dict_cursor.execute(query, (pageid,))
  326. result = dict_cursor.fetchall()[0]
  327. size = self.get_size(content)
  328. m_user, m_time, m_id = self.get_modify(pageid)
  329. if title != result["page_title"]:
  330. self.update_page_title(cursor, result, pageid, title)
  331. if m_id != result["page_modify_oldid"]:
  332. self.update_page_modify(cursor, result, pageid, size, m_user,
  333. m_time, m_id)
  334. if status != result["page_status"]:
  335. special = self.update_page_status(cursor, result, pageid, status,
  336. chart)
  337. s_user = special[0]
  338. else:
  339. s_user = result["page_special_user"]
  340. notes = self.get_notes(chart, content, m_time, s_user)
  341. if notes != result["page_notes"]:
  342. self.update_page_notes(cursor, result, pageid, notes)
  343. def update_page_title(self, cursor, result, pageid, title):
  344. """Update the title of a page in our database."""
  345. query = "UPDATE page SET page_title = ? WHERE page_id = ?"
  346. cursor.execute(query, (title, pageid))
  347. msg = u" {0}: title: {1} -> {2}"
  348. self.logger.debug(msg.format(pageid, result["page_title"], title))
  349. def update_page_modify(self, cursor, result, pageid, size, m_user, m_time, m_id):
  350. """Update the last modified information of a page in our database."""
  351. query = """UPDATE page SET page_size = ?, page_modify_user = ?,
  352. page_modify_time = ?, page_modify_oldid = ?
  353. WHERE page_id = ?"""
  354. cursor.execute(query, (size, m_user, m_time, m_id, pageid))
  355. msg = u" {0}: modify: {1} / {2} / {3} -> {4} / {5} / {6}"
  356. msg = msg.format(pageid, result["page_modify_user"],
  357. result["page_modify_time"],
  358. result["page_modify_oldid"], m_user, m_time, m_id)
  359. self.logger.debug(msg)
  360. def update_page_status(self, cursor, result, pageid, status, chart):
  361. """Update the status and "specialed" information of a page."""
  362. query1 = """UPDATE page JOIN row ON page_id = row_id
  363. SET page_status = ?, row_chart = ? WHERE page_id = ?"""
  364. query2 = """UPDATE page SET page_special_user = ?,
  365. page_special_time = ?, page_special_oldid = ?
  366. WHERE page_id = ?"""
  367. cursor.execute(query1, (status, chart, pageid))
  368. msg = " {0}: status: {1} ({2}) -> {3} ({4})"
  369. self.logger.debug(msg.format(pageid, result["page_status"],
  370. result["row_chart"], status, chart))
  371. s_user, s_time, s_id = self.get_special(pageid, chart)
  372. if s_id != result["page_special_oldid"]:
  373. cursor.execute(query2, (s_user, s_time, s_id, pageid))
  374. msg = u" {0}: special: {1} / {2} / {3} -> {4} / {5} / {6}"
  375. msg = msg.format(pageid, result["page_special_user"],
  376. result["page_special_time"],
  377. result["page_special_oldid"], s_user, s_time, s_id)
  378. self.logger.debug(msg)
  379. return s_user, s_time, s_id
  380. def update_page_notes(self, cursor, result, pageid, notes):
  381. """Update the notes (or warnings) of a page in our database."""
  382. query = "UPDATE page SET page_notes = ? WHERE page_id = ?"
  383. cursor.execute(query, (notes, pageid))
  384. msg = " {0}: notes: {1} -> {2}"
  385. self.logger.debug(msg.format(pageid, result["page_notes"], notes))
  386. def get_content(self, title):
  387. """Get the current content of a page by title from the API.
  388. The page's current revision ID is retrieved from SQL, and then
  389. an API query is made to get its content. This is the only API query
  390. used in the task's code.
  391. """
  392. query = "SELECT page_latest FROM page WHERE page_title = ? AND page_namespace = ?"
  393. try:
  394. namespace, base = title.split(":", 1)
  395. except ValueError:
  396. base = title
  397. ns = wiki.NS_MAIN
  398. else:
  399. try:
  400. ns = self.site.namespace_name_to_id(namespace)
  401. except exceptions.NamespaceNotFoundError:
  402. base = title
  403. ns = wiki.NS_MAIN
  404. result = self.site.sql_query(query, (base.replace(" ", "_"), ns))
  405. try:
  406. revid = int(list(result)[0][0])
  407. except IndexError:
  408. return None
  409. return self.get_revision_content(revid)
  410. def get_revision_content(self, revid, tries=1):
  411. """Get the content of a revision by ID from the API."""
  412. if revid in self.revision_cache:
  413. return self.revision_cache[revid]
  414. res = self.site.api_query(action="query", prop="revisions",
  415. revids=revid, rvprop="content")
  416. try:
  417. content = res["query"]["pages"].values()[0]["revisions"][0]["*"]
  418. except KeyError:
  419. if tries > 0:
  420. sleep(5)
  421. return self.get_revision_content(revid, tries=tries - 1)
  422. self.revision_cache[revid] = content
  423. return content
  424. def get_status_and_chart(self, content, namespace):
  425. """Determine the status and chart number of an AFC submission.
  426. The methodology used here is the same one I've been using for years
  427. (see also commands.afc_report), but with the new draft system taken
  428. into account. The order here is important: if there is more than one
  429. {{AFC submission}} template on a page, we need to know which one to
  430. use (revision history search to find the most recent isn't a viable
  431. idea :P).
  432. """
  433. statuses = self.get_statuses(content)
  434. if "R" in statuses:
  435. status, chart = "r", self.CHART_REVIEW
  436. elif "H" in statuses:
  437. status, chart = "p", self.CHART_DRAFT
  438. elif "P" in statuses:
  439. status, chart = "p", self.CHART_PEND
  440. elif "T" in statuses:
  441. status, chart = None, self.CHART_NONE
  442. elif "D" in statuses:
  443. status, chart = "d", self.CHART_DECLINE
  444. else:
  445. status, chart = None, self.CHART_NONE
  446. if namespace == wiki.NS_MAIN:
  447. if not statuses:
  448. status, chart = "a", self.CHART_ACCEPT
  449. else:
  450. status, chart = None, self.CHART_MISPLACE
  451. return status, chart
  452. def get_statuses(self, content):
  453. """Return a list of all AFC submission statuses in a page's text."""
  454. re_has_templates = "\{\{[aA][fF][cC] submission\s*(\}\}|\||/)"
  455. re_template = "\{\{[aA][fF][cC] submission\s*(.*?)\}\}"
  456. re_remove_embed = "(\{\{[aA][fF][cC] submission\s*(.*?))\{\{(.*?)\}\}(.*?)\}\}"
  457. valid = ["R", "H", "P", "T", "D"]
  458. subtemps = {
  459. "/reviewing": "R",
  460. "/onhold": "H",
  461. "/pending": "P",
  462. "/draft": "T",
  463. "/declined": "D"
  464. }
  465. statuses = []
  466. while re.search(re_has_templates, content):
  467. status = "P"
  468. match = re.search(re_template, content, re.S)
  469. if not match:
  470. return statuses
  471. temp = match.group(1)
  472. limit = 0
  473. while "{{" in temp and limit < 50:
  474. content = re.sub(re_remove_embed, "\\1\\4}}", content, 1, re.S)
  475. match = re.search(re_template, content, re.S)
  476. temp = match.group(1)
  477. limit += 1
  478. params = temp.split("|")
  479. try:
  480. subtemp, params = params[0].strip(), params[1:]
  481. except IndexError:
  482. status = "P"
  483. params = []
  484. else:
  485. if subtemp:
  486. status = subtemps.get(subtemp)
  487. params = []
  488. for param in params:
  489. param = param.strip().upper()
  490. if "=" in param:
  491. key, value = param.split("=", 1)
  492. if key.strip() == "1":
  493. status = value if value in valid else "P"
  494. break
  495. else:
  496. status = param if param in valid else "P"
  497. break
  498. statuses.append(status)
  499. content = re.sub(re_template, "", content, 1, re.S)
  500. return statuses
  501. def get_size(self, content):
  502. """Return a page's size in a short, pretty format."""
  503. return "{0} kB".format(round(len(content) / 1000.0, 1))
  504. def get_modify(self, pageid):
  505. """Return information about a page's last edit ("modification").
  506. This consists of the most recent editor, modification time, and the
  507. lastest revision ID.
  508. """
  509. query = """SELECT rev_user_text, rev_timestamp, rev_id FROM revision
  510. JOIN page ON rev_id = page_latest WHERE page_id = ?"""
  511. result = self.site.sql_query(query, (pageid,))
  512. m_user, m_time, m_id = list(result)[0]
  513. timestamp = datetime.strptime(m_time, "%Y%m%d%H%M%S")
  514. return m_user.decode("utf8"), timestamp, m_id
  515. def get_special(self, pageid, chart):
  516. """Return information about a page's "special" edit.
  517. I tend to use the term "special" as a verb a lot, which is bound to
  518. cause confusion. It is merely a short way of saying "the edit in which
  519. a declined submission was declined, an accepted submission was
  520. accepted, a submission in review was set as such, a pending submission
  521. was submitted, and a "misplaced" submission was created."
  522. This "information" consists of the special edit's editor, its time, and
  523. its revision ID. If the page's status is not something that involves
  524. "special"-ing, we will return None for all three. The same will be
  525. returned if we cannot determine when the page was "special"-ed, or if
  526. it was "special"-ed more than 100 edits ago.
  527. """
  528. if chart == self.CHART_NONE:
  529. return None, None, None
  530. elif chart == self.CHART_MISPLACE:
  531. return self.get_create(pageid)
  532. elif chart == self.CHART_ACCEPT:
  533. search_for = None
  534. search_not = ["R", "H", "P", "T", "D"]
  535. elif chart == self.CHART_DRAFT:
  536. search_for = "H"
  537. search_not = []
  538. elif chart == self.CHART_PEND:
  539. search_for = "P"
  540. search_not = []
  541. elif chart == self.CHART_REVIEW:
  542. search_for = "R"
  543. search_not = []
  544. elif chart == self.CHART_DECLINE:
  545. search_for = "D"
  546. search_not = ["R", "H", "P", "T"]
  547. query = """SELECT rev_user_text, rev_timestamp, rev_id
  548. FROM revision WHERE rev_page = ? ORDER BY rev_id DESC"""
  549. result = self.site.sql_query(query, (pageid,))
  550. counter = 0
  551. last = (None, None, None)
  552. for user, ts, revid in result:
  553. counter += 1
  554. if counter > 50:
  555. msg = "Exceeded 50 content lookups while determining special for page (id: {0}, chart: {1})"
  556. self.logger.warn(msg.format(pageid, chart))
  557. return None, None, None
  558. try:
  559. content = self.get_revision_content(revid)
  560. except exceptions.APIError:
  561. msg = "API error interrupted SQL query in get_special() for page (id: {0}, chart: {1})"
  562. self.logger.exception(msg.format(pageid, chart))
  563. return None, None, None
  564. statuses = self.get_statuses(content)
  565. matches = [s in statuses for s in search_not]
  566. if search_for:
  567. if search_for not in statuses or any(matches):
  568. return last
  569. else:
  570. if any(matches):
  571. return last
  572. timestamp = datetime.strptime(ts, "%Y%m%d%H%M%S")
  573. last = (user.decode("utf8"), timestamp, revid)
  574. return last
  575. def get_create(self, pageid):
  576. """Return information about a page's first edit ("creation").
  577. This consists of the page creator, creation time, and the earliest
  578. revision ID.
  579. """
  580. query = """SELECT rev_user_text, rev_timestamp, rev_id
  581. FROM revision WHERE rev_id =
  582. (SELECT MIN(rev_id) FROM revision WHERE rev_page = ?)"""
  583. result = self.site.sql_query(query, (pageid,))
  584. c_user, c_time, c_id = list(result)[0]
  585. timestamp = datetime.strptime(c_time, "%Y%m%d%H%M%S")
  586. return c_user.decode("utf8"), timestamp, c_id
  587. def get_notes(self, chart, content, m_time, s_user):
  588. """Return any special notes or warnings about this page.
  589. copyvio: submission is a suspected copyright violation
  590. unsourced: submission lacks references completely
  591. no-inline: submission has no inline citations
  592. short: submission is less than a kilobyte in length
  593. resubmit: submission was resubmitted after a previous decline
  594. old: submission has not been touched in > 4 days
  595. blocked: submitter is currently blocked
  596. """
  597. notes = ""
  598. ignored_charts = [self.CHART_NONE, self.CHART_ACCEPT, self.CHART_DECLINE]
  599. if chart in ignored_charts:
  600. return notes
  601. copyvios = self.config.tasks.get("afc_copyvios", {})
  602. regex = "\{\{\s*" + copyvios.get("template", "AfC suspected copyvio")
  603. if re.search(regex, content):
  604. notes += "|nc=1" # Submission is a suspected copyvio
  605. if not re.search("\<ref\s*(.*?)\>(.*?)\</ref\>", content, re.I | re.S):
  606. regex = "(https?:)|\[//(?!{0})([^ \]\\t\\n\\r\\f\\v]+?)"
  607. sitedomain = re.escape(self.site.domain)
  608. if re.search(regex.format(sitedomain), content, re.I | re.S):
  609. notes += "|ni=1" # Submission has no inline citations
  610. else:
  611. notes += "|nu=1" # Submission is completely unsourced
  612. if len(content) < 1000:
  613. notes += "|ns=1" # Submission is short
  614. statuses = self.get_statuses(content)
  615. if "D" in statuses and chart != self.CHART_MISPLACE:
  616. notes += "|nr=1" # Submission was resubmitted
  617. time_since_modify = (datetime.utcnow() - m_time).total_seconds()
  618. max_time = 4 * 24 * 60 * 60
  619. if time_since_modify > max_time:
  620. notes += "|no=1" # Submission hasn't been touched in over 4 days
  621. if chart in [self.CHART_PEND, self.CHART_DRAFT] and s_user:
  622. submitter = self.site.get_user(s_user)
  623. try:
  624. if submitter.blockinfo:
  625. notes += "|nb=1" # Submitter is blocked
  626. except exceptions.UserNotFoundError: # Likely an IP
  627. pass
  628. return notes