PS1: shorten slightly & add PS0 recovery

[home.git] / test.py

diff --git a/test.py b/test.py
index 2bbef324317c732020b5cea5fe6a3cdcaba78ff9..e6436150e4d1029be55be20f5ab59acb08d84850 100755 (executable)
--- a/test.py
+++ b/test.py
@@ -4,26 +4,48 @@
 """Test module for quick hacking."""
 
 from __future__ import absolute_import
+from __future__ import division
 from __future__ import print_function
 
+# pylint: disable=unused-import
 import argparse
 import atexit
+import codecs
 import datetime
 import errno
+import fileinput
 import fnmatch
 import glob
-import mock
+import io
 import multiprocessing
 import os
 import re
-import requests
 import signal
 import subprocess
 import sys
 import time
+import unittest
+
+try:
+    from unittest import mock
+except ImportError:
+    import mock
+
+import requests
+import six
+
+
+def get_parser():
+    """Get CLI parser."""
+    parser = argparse.ArgumentParser(description=__doc__)
+    return parser
 
 
 def main(argv):
+    """The main entry point for scripts."""
+    parser = get_parser()
+    opts = parser.parse_args(argv)
+
     print(argv)