X-Git-Url: https://git.wh0rd.org/?a=blobdiff_plain;f=test.py;h=9edf5ddc0069888783a6b64847662c55a59d4c2b;hb=6ba00e98257c3777c63600b1af5356d5fb73f881;hp=70d4ffbe69e0dc31b80b971d4df76d6de7eb599a;hpb=b6ccd09e8e78866264be7df044ac10f78a1d5002;p=home.git

diff --git a/test.py b/test.py
index 70d4ffb..9edf5dd 100755
--- a/test.py
+++ b/test.py
@@ -4,33 +4,56 @@
 """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 abc, argparse, atexit
+import base64, binascii
+import codecs, collections, contextlib
 import datetime
 import errno
-import fileinput
-import fnmatch
+import fcntl, fileinput, fnmatch
 import glob
-import io
+import hashlib
+import inspect, io, itertools
+import json
+import logging
 import multiprocessing
 import os
+import pprint
 import re
-import signal
-import subprocess
-import sys
-import time
+import shlex, shutil, signal, socket, struct, subprocess, sys
+import tarfile, tempfile, time, traceback
+import unittest
+
+if sys.version_info.major < 3:
+    import mock
+else:
+    import pathlib
+    from pathlib import Path
+    import typing
+    from typing import (
+        Any, Callable, Dict, Generator, Iterable, Iterator, List, NamedTuple,
+        Optional, Set, Tuple, Union,
+    )
+    from unittest import mock
 
-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)