🐍🌚🤖 Formatted using black.

Author: mariatta-bot

path

@@ -0,0 +1,157 @@
+import celery
+import base64
+import os
+import subprocess
+
+from celery import bootsteps
+
+from . import util, exceptions
+
+app = celery.Celery("black_out")
+
+app.conf.update(
+    BROKER_URL=os.environ["REDIS_URL"], CELERY_RESULT_BACKEND=os.environ["REDIS_URL"]
+)
+
+
+@app.task(rate_limit="1/m")
+def setup_repo():
+
+    repo_name = os.environ.get("GH_REPO_NAME")
+    repo_full_name = os.environ.get("GH_REPO_FULL_NAME")
+    os.mkdir("repo_checkout")
+    os.chdir("repo_checkout")
+    print(f"Setting up {repo_name} repository, cloning from {repo_full_name}")
+
+    if repo_name not in os.listdir("."):
+        email_address = os.environ.get("GH_EMAIL")
+        full_name = os.environ.get("GH_FULL_NAME")
+        subprocess.check_output(
+            [
+                "git",
+                "clone",
+                f"https://{os.environ.get('GH_AUTH')}:x-oauth-basic@github.com/{repo_full_name}.git",
+            ]
+        )
+        subprocess.check_output(
+            ["git", "config", "--global", "user.email", f"'{email_address}'"]
+        )
+        subprocess.check_output(
+            ["git", "config", "--global", "user.name", f"'{full_name}'"]
+        )
+        os.chdir(f"./{repo_name}")
+        print(f"Finished setting up {repo_name} Repo")
+    else:
+        print(f"{repo_name} directory already exists")
+
+
+@app.task(rate_limit="1/m")
+def initiate_black_task(issue_number, issue_creator):
+    """Execute black
+
+    1. git fetch origin
+    2. git checkout -b issue-NNNN-initialize-black origin/master
+    3. black .
+    4. git commit -am "🤖 Format code using `black` ..."
+    5. git push origin issue-NNNN-initialize-black
+    6. create the PR
+    7. git checkout master
+    7. git branch -D issue-NNNN-initialize-black
+    """
+    # cd to the checked out repo, if not already there
+    if "repo_checkout" in os.listdir("."):
+        os.chdir("repo_checkout")
+        os.chdir(f"./{os.environ.get('GH_REPO_NAME')}")
+
+    branch_name = f"issue-{issue_number}-initialize-black"
+
+    util.exec_command(["git", "fetch", "origin"])
+
+    try:
+        util.checkout_branch(branch_name)
+    except exceptions.BranchCheckoutException as e:
+        message = f"""
+🤖 Sorry @{issue_creator}. I was not able to check out the branch in order
+to create the pull request. Perhaps a pull request has been made, or
+black has been initiated? (I'm a bot 🤖)
+"""
+        util.comment_on_pr(issue_number, message)
+        raise e
+
+    needs_black = util.check_black(["."])
+    if needs_black:
+        util.exec_command(["black", "."])
+        commit_title, commit_body = util.commit_changes(issue_number)
+        util.exec_command(["git", "push", "origin", branch_name])
+        util.create_gh_pr("master", branch_name, title=commit_title, body=commit_body)
+        util.exec_command(["git", "checkout", "master"])
+        util.delete_branch(branch_name)
+    else:
+        message = f"""
+🤖 @{issue_creator}, the repo appears to be already well formatted with `black`.
+Closing the issue. 🌮
+(I'm a bot 🤖)
+"""
+
+        util.comment_on_pr(issue_number, message)
+        util.close_issue(issue_number)
+
+
+@app.task(rate_limit="1/m")
+def black_pr_task(pr_number, pr_author, pr_diff_url):
+    """Execute black on a PR
+
+    1. git fetch origin pull/{pr_number}/head:pr_{pr_number} && git checkout pr_number
+    2. find out all affected files
+    3. black <all affected files>
+    4. git commit -am "🤖 Format code using `black` ..."
+    5. git push git@github.com:<pr_author>/{repo_name} pr_{pr_number}:{branch_name}
+    6. comment on PR
+    7. git checkout master
+    8. git branch -D pr_{pr_number}
+    """
+    # cd to the checked out repo, if not already there
+    if "repo_checkout" in os.listdir("."):
+        os.chdir("repo_checkout")
+        os.chdir(f"./{os.environ.get('GH_REPO_NAME')}")
+
+    util.exec_command(
+        ["git", "fetch", "origin", f"pull/{pr_number}/head:pr_{pr_number}"]
+    )
+    util.exec_command(["git", "checkout", f"pr_{pr_number}"])
+    files_affected = util.get_pr_diff_files(pr_diff_url)
+    branch_name = f"pr_{pr_number}"
+
+    needs_black = util.check_black(files_affected)
+
+    if needs_black:
+        commands = ["black"]
+        commands.extend(files_affected)
+        util.exec_command(commands)
+
+        commit_title, commit_body = util.commit_changes(pr_number)
+        util.exec_command(
+            [
+                "git",
+                "push",
+                f"git@github.com:<{pr_author}>/{os.environ.get('GH_REPO_NAME')}",
+                branch_name,
+            ]
+        )
+        util.create_gh_pr("master", branch_name, title=commit_title, body=commit_body)
+        message = f"""
+🤖 @{pr_author}, I've reformatted the code using `black` for you. 🌮
+(I'm a bot 🤖)
+"""
+        util.comment_on_pr(pr_number, message)
+    util.exec_command(["git", "checkout", "master"])
+    util.delete_branch(branch_name)
+
+
+class InitRepoStep(bootsteps.StartStopStep):
+    def start(self, c):
+        print("Initialize the repository.")
+        setup_repo()
+
+
+app.steps["worker"].add(InitRepoStep)