Rewrite ./site.sh to Python

site.py

@@ -0,0 +1,101 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+import argparse
+import os
+import subprocess
+import sys
+import shlex
+from datetime import datetime
+
+class Logger:
+
+    @classmethod
+    def print_out(cls, msg: str, end=os.linesep):
+        print(msg, end=end, file=sys.stdout)
+
+    @classmethod
+    def print_err(cls, msg: str, end=os.linesep):
+        print(msg, end=end, file=sys.stderr)
+
+print_out = Logger.print_out
+print_err = Logger.print_err
+
+def run(command: str, check=True, silent=False) -> subprocess.CompletedProcess:
+    """
+    Runs a command provided as a string
+
+    :param command: Command provided as a string
+    :param check: If true, will throw exception on failure
+    :param silent: If set to True, the output won't be printed.
+
+    :return: CompleteProcess instance - the result of the command executed
+    """
+
+    if not silent:
+        # https://docs.python.org/3/library/datetime.html#examples-of-usage-datetime
+        log_msg = f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] " \
+                  f"Executing: {command}" + os.linesep
+        print_out(log_msg)
+        print_err(log_msg)
+
+    # https://docs.python.org/3/library/subprocess.html
+    proc = subprocess.run(shlex.split(command), check=check, stderr=subprocess.PIPE, stdout=subprocess.PIPE)
+
+    if not silent:
+        print_err(proc.stderr.decode())
+        print_out(proc.stdout.decode())
+
+    return proc
+
+
+def build_site():
+    run('which git')
+
+def build_image():
+    run('docker build -t airflow-site .')
+
+def parse_args():
+    parser = argparse.ArgumentParser(description='Build Airflow website')
+    parser.add_argument("action", choices=['build-site', 'build-image', 'stop'], default="build-site", nargs="?")
+
+    args = parser.parse_args()
+    return args
+
+
+def main():
+    """
+    Main function of the script
+    """
+    args = parse_args()
+
+    if args.action == 'build-site':
+        build_site()
+    elif args.action == 'build-image':
+        build_image()
+    elif args.action == 'stop':
+        print("do nothing")
+        sys.exit(1)
+
+
+
+if __name__ == '__main__':
+    try:
+        main()
+    except Exception as err:
+        print("Script failed to run")
+        raise err