updated syntax using Black

Author: TimidRobot

tools/add_cc0_links.py

@@ -17,25 +17,28 @@
 import re, sys
 from pathlib import Path
 
-class AddCC0Links(object):
 
+class AddCC0Links(object):
     def usage(self):
-        print('add-cc0-links.py LANGUAGE_CODE LANGUAGE_NAME')
-        print('    e.g. add-cc0-links.py nl Nederlands')
-        print('    LANGUAGE_CODE must be 2 letters or 2-hyphen-N, the same used in filename.')
-        print('    LANGUAGE_NAME must be in the relevant language')
-        print('                  if it contains whitespace, enclose in quotes.')
+        print("add-cc0-links.py LANGUAGE_CODE LANGUAGE_NAME")
+        print("    e.g. add-cc0-links.py nl Nederlands")
+        print(
+            "    LANGUAGE_CODE must be 2 letters or 2-hyphen-N, the same used in filename."
+        )
+        print("    LANGUAGE_NAME must be in the relevant language")
+        print("                  if it contains whitespace, enclose in quotes.")
 
     def get_args(self):
         # Make sure there are enough args
         # Make sure arg 2 is a language code
         # Make sure arg 3 is not a language code
-        self.args_ok = (len(sys.argv) == 3) and (len(sys.argv[1]) >= 2) \
-                       and (len(sys.argv[2]) >= 2)
+        self.args_ok = (
+            (len(sys.argv) == 3) and (len(sys.argv[1]) >= 2) and (len(sys.argv[2]) >= 2)
+        )
         if self.args_ok:
             self.language_code = sys.argv[1]
             self.language_name = sys.argv[2]
-            self.exclude_pattern = 'zero_1.0_' + self.language_code + '.html'
+            self.exclude_pattern = "zero_1.0_" + self.language_code + ".html"
         else:
             self.usage()
         return self.args_ok
@@ -45,20 +48,23 @@ def get_path(self):
         self.path = False
         path = Path.cwd()
         pathdir = path.name
-        if pathdir == 'legalcode':
+        if pathdir == "legalcode":
             self.path = path
-        if pathdir == 'docroot':
-            self.path = path / 'legalcode'
-        if pathdir == 'tools':
-            self.path = path.parent / 'docroot' /'legalcode'
+        if pathdir == "docroot":
+            self.path = path / "legalcode"
+        if pathdir == "tools":
+            self.path = path.parent / "docroot" / "legalcode"
         if not self.path:
-            print('Please run from within the checked-out project.')
+            print("Please run from within the checked-out project.")
         return self.path != False
 
     def get_files(self):
         """Get all the CC0 files *except* those we are linking to"""
-        self.files = [f for f in self.path.glob('zero_1.0*.html')
-                      if not f.match(self.exclude_pattern)]
+        self.files = [
+            f
+            for f in self.path.glob("zero_1.0*.html")
+            if not f.match(self.exclude_pattern)
+        ]
 
     def process_files(self):
         """Add links to all the license files"""
@@ -67,15 +73,18 @@ def process_files(self):
 
     def file_license_and_language(self, filepath):
         """Get the license number and language code from the file path"""
-        elements = filepath.stem.split('_')
+        elements = filepath.stem.split("_")
         # Un-translated deeds don't have a language code, so set to English
         if len(elements) != 3:
-            elements += ['en']
+            elements += ["en"]
         return elements[0], elements[2]
 
     def links_in_page(self, content):
         """Find the translated license links at the bottom of the page"""
-        return re.findall(r'//creativecommons\.org/publicdomain/zero/1\.0/legalcode(\...)?">([^>]+)</a>', content)
+        return re.findall(
+            r'//creativecommons\.org/publicdomain/zero/1\.0/legalcode(\...)?">([^>]+)</a>',
+            content,
+        )
 
     def is_rtl(self, content):
         """Determine whether the page is in a right-to-left script"""
@@ -98,22 +107,35 @@ def insert_link(self, content, lic, links, index):
         """Insert the link to the correct version of the license
            in the correct position in the list of links at the bottom of the
            page"""
-        link = '<a href="//creativecommons.org/publicdomain/zero/1.0/legalcode.' + self.language_code + '">' + self.language_name + '</a>'
+        link = (
+            '<a href="//creativecommons.org/publicdomain/zero/1.0/legalcode.'
+            + self.language_code
+            + '">'
+            + self.language_name
+            + "</a>"
+        )
         if index == -1:
             target = '<a href="//creativecommons.org/publicdomain/zero/1.0/'
-            replace = link + ', ' + target
+            replace = link + ", " + target
         else:
             lang = links[index][1]
-            target = '>' + lang + '</a>'
-            replace = target + ', ' + link
+            target = ">" + lang + "</a>"
+            replace = target + ", " + link
         return content.replace(target, replace, 1)
 
     def file_contains_link_already(self, links):
         """Did we already add a link to this page?"""
-        return next((code for code, name in links
-                     if name == self.language_name
-                     or code == self.language_code),
-                    False) != False
+        return (
+            next(
+                (
+                    code
+                    for code, name in links
+                    if name == self.language_name or code == self.language_code
+                ),
+                False,
+            )
+            != False
+        )
 
     def process_file(self, filepath):
         """Get the file's details and insert a link to the translated version
@@ -130,18 +152,19 @@ def process_file(self, filepath):
             print(index)
             print(links[index])
             updated_content = self.insert_link(content, lic, links, index)
-            with filepath.open('w') as outfile:
+            with filepath.open("w") as outfile:
                 outfile.write(updated_content)
-            print('Added link to file: ' + filepath.name)
+            print("Added link to file: " + filepath.name)
         else:
-            print('File already contains link: ' + filepath.name)
+            print("File already contains link: " + filepath.name)
 
     def main(self):
         """Get the command line arguments, find the files, and process them"""
         if self.get_args() and self.get_path():
             self.get_files()
             self.process_files()
 
-if __name__ == '__main__':
+
+if __name__ == "__main__":
     link_adder = AddCC0Links()
     link_adder.main()