Formatting

Author: canepan

src/tools/bin/arp_map.py

@@ -13,14 +13,26 @@
 class ArpMapRecord(object):
     seen_names: dict = {}
 
-    def __init__(self, fqdn: str = None, ip: str = None, mac: str = None, interface: str = None, known_hosts_styler: callable = None, duplicate_hosts_styler: callable = None):
+    def __init__(
+        self,
+        fqdn: str = None,
+        ip: str = None,
+        mac: str = None,
+        interface: str = None,
+        known_hosts_styler: callable = None,
+        duplicate_hosts_styler: callable = None,
+    ):
         self.fqdn = fqdn
         self.ip = ip
         self.mac = mac
         self.interface = interface
         self.name = self.fqdn if self.fqdn != "?" else self.ip
-        self.known_hosts_styler = known_hosts_styler if known_hosts_styler is not None else lambda x: click.style(x, 'green')
-        self.duplicate_hosts_styler = duplicate_hosts_styler if duplicate_hosts_styler is not None else lambda x: click.style(x, 'red')
+        self.known_hosts_styler = (
+            known_hosts_styler if known_hosts_styler is not None else lambda x: click.style(x, 'green')
+        )
+        self.duplicate_hosts_styler = (
+            duplicate_hosts_styler if duplicate_hosts_styler is not None else lambda x: click.style(x, 'red')
+        )
         if self.name in ArpMapRecord.seen_names:
             ArpMapRecord.seen_names[self.name] = self.duplicate_hosts_styler(self.name)
         else:
@@ -103,7 +115,7 @@ def __iter__(self) -> typing.Iterable[ArpMapRecord]:
                 yield record
 
 
-def add_record_to_list(record: ArpMapRecord, records_list: list) ->  None:
+def add_record_to_list(record: ArpMapRecord, records_list: list) -> None:
     if record.fqdn.split('.')[0].lower() in KNOWN_HOSTS:
         records_list.insert(0, record)
     else:

src/tools/bin/backup_mysql.py

@@ -11,10 +11,10 @@
 
 class GzipStream(object):
     # input is a filelike object that feeds the input
-    def __init__(self, input, filename = None):
+    def __init__(self, input, filename=None):
         self.input = input
         self.buffer = b''
-        self.zipper = GzipFile(filename, mode = 'wb', fileobj = self)
+        self.zipper = GzipFile(filename, mode='wb', fileobj=self)
 
     def read(self, size=-1):
         if (size < 0) or len(self.buffer) < size:
@@ -43,7 +43,9 @@ def close(self):
 
 
 def parse_args(argv=sys.argv[1:]):
-    parser = argparse.ArgumentParser(description='Backup all mysql DBs', formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+    parser = argparse.ArgumentParser(
+        description='Backup all mysql DBs', formatter_class=argparse.ArgumentDefaultsHelpFormatter
+    )
     parser.add_argument('--output-file', default='/mnt/opt/backup/mysql/{}_all_dump.sql.gz'.format(os.uname().nodename))
     parser.add_argument('--mysql-socket', default='/var/run/mysqld/mysqld.sock')
     parser.add_argument('--debug', default='INFO', choices=('INFO', 'DEBUG', 'WARN', 'ERROR'))
@@ -69,7 +71,7 @@ def new_size(self) -> int:
             self._new_size = os.stat(self.new_file).st_size
         return self._new_size
 
-    def size_difference_percent(self)-> float:
+    def size_difference_percent(self) -> float:
         return (self.new_size - self.old_size) / self.old_size
 
 
@@ -79,7 +81,9 @@ def main(argv: list = sys.argv[1:]):
     _log.addHandler(logging.StreamHandler(sys.stdout))
     _log.setLevel(cfg.debug)
 
-    _output = tempfile.NamedTemporaryFile(delete=False, dir=os.path.dirname(cfg.output_file), prefix=os.path.basename(cfg.output_file))
+    _output = tempfile.NamedTemporaryFile(
+        delete=False, dir=os.path.dirname(cfg.output_file), prefix=os.path.basename(cfg.output_file)
+    )
     _log.info('Running mysqldump')
     command = subprocess.Popen(['mysqldump', '-A', '-S', cfg.mysql_socket], stdout=subprocess.PIPE)
     gzipstream = GzipStream(command.stdout)

src/tools/bin/compare_ka.py

@@ -9,12 +9,18 @@
 
 
 @click.command()
-@click.argument("hostnames", nargs=-1) #, type=set)
+@click.argument("hostnames", nargs=-1)  # , type=set)
 def main(hostnames: set):
     hc = HostChecker(hostnames)
     for host in hosts_if_not_me([h.name for h in hc.hosts]):
         click.echo(f"Checking {host}...")
-        click.echo(check_output(["remote_diff.sh", "/etc/keepalived/keepalived.d/", f"{host.hostname}:/etc/keepalived/keepalived.d/"], universal_newlines=True, stderr=STDOUT))
+        click.echo(
+            check_output(
+                ["remote_diff.sh", "/etc/keepalived/keepalived.d/", f"{host.hostname}:/etc/keepalived/keepalived.d/"],
+                universal_newlines=True,
+                stderr=STDOUT,
+            )
+        )
 
 
 if __name__ == "__main__":

src/tools/bin/openvpn_log.py

@@ -21,9 +21,9 @@ def main(argv: list = sys.argv[1:]) -> int:
         with open(log_file, 'r') as f:
             lines = f.readlines()
         for line in lines:
-            uniques[tuple(
-                re.sub(r':[0-9]+', ':<port>', w) for w in line.split()[5:] if not re.search(skip_regexp, line)
-            )] = line
+            uniques[
+                tuple(re.sub(r':[0-9]+', ':<port>', w) for w in line.split()[5:] if not re.search(skip_regexp, line))
+            ] = line
         print(''.join(sorted(uniques.values())))
     return skipped