diff options
-rw-r--r-- | salvager/__init__.py (renamed from scrap.py) | 8 | ||||
-rw-r--r-- | salvager/__main__.py | 8 |
2 files changed, 10 insertions, 6 deletions
diff --git a/scrap.py b/salvager/__init__.py index c430f36..0b19f16 100644 --- a/scrap.py +++ b/salvager/__init__.py @@ -4,8 +4,8 @@ import re import sys import time -from salvager import esi -from salvager import zkill +from . import esi +from . import zkill SYSTEM_AND_DATE = re.compile(r"/(\d+)/(\d+)/?$") @@ -101,7 +101,3 @@ def main(): opts = {'indent': 4} if args.pretty else {} fd.write(json.dumps(snapshot, **opts)) fd.write("\n") - - -if __name__ == "__main__": - main() diff --git a/salvager/__main__.py b/salvager/__main__.py new file mode 100644 index 0000000..05769a5 --- /dev/null +++ b/salvager/__main__.py @@ -0,0 +1,8 @@ +import sys + +from . import main + +if __name__ == "__main__": + if "__main__" in sys.argv[0]: + sys.argv[0] = "salvager" + main() |