User:Lowercase sigmabot II/Source.py: Difference between revisions

Revision as of 19:32, 18 July 2020

Latest revision as of 17:16, 8 May 2022

(One intermediate revision by one other user not shown)

Line 17:

If the template is present, it does nothing.

<sourcesyntaxhighlight lang="python">

#!/home/sigma/.local/bin/python3

# -*- coding: utf-8 -*-

Line 85:

main()

</syntaxhighlight>

</source>

== Sandbot2.py ==

Line 97:

* Feel free to try your editing skills below *

■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■--></pre>

<syntaxhighlight lang="python">

#!/home/sigma/.local/bin/python3

# -*- coding: utf-8 -*-

# LGPLv2+ license, look it up

import time

import os

import sys

import pickle

import threading

import arrow

import ceterach

import passwords

reset_text = "{{subst:Sandbox reset}}"

tpl_reset_text = "{{subst:Template sandbox reset}}"

def main():

global api

api = ceterach.api.MediaWiki("https://en.wikipedia.org/w/api.php")

with open("opener.pkl", 'rb') as fp:

api.opener = pickle.load(fp)

bot = SandBot2(api)

bot.run()

class SandBot2:

SANDBOXES = ("Wikipedia:Sandbox",

"Wikipedia:Tutorial/Editing/sandbox",

"Wikipedia:Tutorial/Formatting/sandbox",

"Wikipedia:Tutorial/Wikipedia links/sandbox",

"Wikipedia:Tutorial/Citing sources/sandbox",

"Wikipedia:Tutorial/Keep in mind/sandbox"

)

SANDBOXES = {"Wikipedia:Sandbox",

"Wikipedia:Tutorial/Editing/sandbox",

"Wikipedia:Tutorial/Formatting/sandbox",

"Wikipedia:Tutorial/Wikipedia links/sandbox",

"Wikipedia:Tutorial/Citing sources/sandbox",

"Wikipedia:Tutorial/Keep in mind/sandbox",

"Wikipedia talk:Sandbox",

"Wikipedia talk:Tutorial/Editing/sandbox",

"Wikipedia talk:Tutorial/Formatting/sandbox",

"Wikipedia talk:Tutorial/Wikipedia links/sandbox",

"Wikipedia talk:Tutorial/Citing sources/sandbox",

"Wikipedia talk:Tutorial/Keep in mind/sandbox",

"User:Sandbox",

"Template:Template sandbox",

}

def __init__(self, api, shutoff="User:Lowercase sigmabot II/Shutoff"):

self.api = api

self.shutoff_page = api.page(shutoff)

@property

def is_allowed(self):

return self.shutoff_page.content.lower() == "true"

def wait(self, box):

for __ in range(3):

# Sleep for 3 minutes

print("3 minute sleep on {!r}".format(box.title))

time.sleep(60 * 3)

if self.box_needs_reset(box):

break

# After the bot sleeps on this box for 9 minutes, it

# will clear the box regardless.

print("Done with sleeping, clearing {!r}".format(box.title))

self.api.login("Lowercase sigmabot II", passwords.lcsb2)

self.api.set_token("edit")

text = tpl_reset_text if box.namespace == 10 else reset_text

box.edit(text, "Clearing sandbox) (bot", bot=True, force=True)

def parse_date(self, date):

return arrow.Arrow.strptime(date, '%Y-%m-%dT%H:%M:%SZ')

def box_needs_reset(self, box):

now = arrow.utcnow()

three_min = arrow.util.timedelta(seconds=60 * 3)

# there's probably a way to use MediaWiki.iterator(), but too lazy

res = self.api.call(action="query", prop="revisions", titles=box.title, rvprop="timestamp", limit="2")

str_stamp = next(iter(res["query"]["pages"].values()))["revisions"][0]["timestamp"]

box_stamp = self.parse_date(str_stamp)

if box_stamp < (now - three_min):

return True

return False

def run(self):

if not self.is_allowed:

print("Check the shutoff page")

return

for sandbox in self.SANDBOXES:

box = self.api.page(sandbox)

if box.revision_user.name in ("Lowercase sigmabot II", "Hazard-Bot") or\

box.content == reset_text:

continue

if self.box_needs_reset(box):

print("Clearing {!r}".format(sandbox))

print(box.edit(reset_text, "Clearing sandbox) (bot", bot=True, force=True))

else:

# The bot will fork, and continue on to the next sandbox

# while the child process waits 3 to 9 minutes

# if os.fork() == 0: # Child process

# self.wait(box) # Wait takes place in the child process

# os._exit(0) # Exit the child process

threading.Thread(target=self.wait, args=(box,)).start()

if __name__ == "__main__":

main()

</syntaxhighlight>