diff --git a/samples/amcache.py b/samples/amcache.py index e816b93..8d340a8 100644 --- a/samples/amcache.py +++ b/samples/amcache.py @@ -220,6 +220,8 @@ def main(argv=None): for e in ee: w.writerow(map(lambda i: getattr(e, i.name), FIELDS)) +def _main(): + main(argv=sys.argv) if __name__ == "__main__": - main(argv=sys.argv) + main() diff --git a/samples/forensicating.py b/samples/forensicating.py index 574c1f7..cb65e1d 100755 --- a/samples/forensicating.py +++ b/samples/forensicating.py @@ -300,7 +300,7 @@ def users_info(soft_reg): print(line) -if __name__ == "__main__": +def main(): """ Print out all of the information """ @@ -317,3 +317,7 @@ def users_info(soft_reg): network_settings(sys_reg, soft_reg) users_info(soft_reg) user_reg_locs(users_paths(soft_reg, users_sids(soft_reg))) + +if __name__ == "__main__": + main() + diff --git a/setup.py b/setup.py index 89d08ae..f329db4 100755 --- a/setup.py +++ b/setup.py @@ -15,6 +15,13 @@ "Programming Language :: Python :: 3", "Operating System :: OS Independent", "License :: OSI Approved :: Apache Software License"], - install_requires=['enum-compat'] + install_requires=['enum-compat'], + entry_points = { + "console_scripts": [ + "forensicating.py = samples.forensicating:main", + "amcache.py = samples.amcache:main", + "registry-timeline.py = samples.timeline:main", + ] + } )