Toggles input with file line in sublime text. Useful for competative programming

ToggleFileInput.py

import sublime_plugin


class TogglePythonFileInputCommand(sublime_plugin.TextCommand):
    def run(self, edit):
        reg1 = self.view.find(r"# sys\.stdin = open\('in', 'r'\)", 0)
        reg2 = self.view.find(r"# import sys", 0)

        if not reg1.empty():
            self.view.replace(edit, reg1, "sys.stdin = open('in', 'r')")
        else:
            reg1 = self.view.find(r"sys\.stdin = open\('in', 'r'\)", 0)
            if not reg1.empty():
                self.view.replace(edit, reg1, "# sys.stdin = open('in', 'r')")

        if not reg2.empty():
            self.view.replace(edit, reg2, "import sys")
        else:
            reg2 = self.view.find(r'import sys', 0)
            if not reg2.empty():
                self.view.replace(edit, reg2, "# import sys")