HW5(Yurin Andrey)

homeworks/Yurin Andrey/№5/my-awesome-script/my_awesome_script/__init__.py

@@ -0,0 +1 @@
+

homeworks/Yurin Andrey/№5/my-awesome-script/my_awesome_script/__main__.py

@@ -0,0 +1,17 @@
+from .cmdInput import cmd_input
+
+
+def main():
+    try:
+        cmd_input()
+    except Exception as ex:
+        print('You have done something wrong!', ex)
+
+
+if __name__ == '__main__':
+    try:
+        main()
+    except KeyboardInterrupt:
+        print()
+        print('Shutting down, bye!')
+

homeworks/Yurin Andrey/№5/my-awesome-script/my_awesome_script/cmdInput.py

@@ -0,0 +1,17 @@
+import argparse
+
+from .command import command, functions
+
+
+def cmd_input():
+    """Input method."""
+    parser = argparse.ArgumentParser()
+    parser.add_argument('operation', help='Input command', choices=functions)
+    parser.add_argument('parameter', help="'Any text' or 'Region/City'")
+    args = parser.parse_args()
+
+    try:
+        command(args.operation, args.parameter)
+    except Exception as ex:
+        command('help')
+        raise ex

homeworks/Yurin Andrey/№5/my-awesome-script/my_awesome_script/command.py

@@ -0,0 +1,35 @@
+
+from datetime import datetime
+
+import pytz
+from cowpy import cow
+from pygments import highlight
+from pygments.formatters import TerminalFormatter
+from pygments.lexers import PythonLexer
+
+
+def command(operation: str, text: str = None) -> None:
+    functions[operation]() if operation == 'help' else functions[operation](text)
+
+
+def highlight_(code: str) -> None:
+    print(highlight(code, PythonLexer(), TerminalFormatter()))
+
+
+def cowsay_(text: str) -> None:
+    print(cow.milk_random_cow(text))
+
+
+def time_(timezone: str) -> None:
+    local_date = datetime.now(pytz.timezone(timezone))
+    print("{0} {1}".format(local_date.date(), local_date.time()))
+
+
+def help_() -> None:
+    print('EXAMPLE COMMAND:')
+    print(" highlight 'any text'")
+    print(" cowsay 'any text'")
+    print(" time 'Region/City' (example: Europe/Moscow)")
+
+
+functions = {'highlight': highlight_, 'cowsay': cowsay_, 'time': time_, 'help': help_}