Index: setup.py
===================================================================
--- setup.py (revision 444)
+++ setup.py (working copy)
@@ -35,6 +35,7 @@
entry_points = """
[console_scripts]
migrate = migrate.versioning.shell:main
+ migrate-repository = migrate.versioning.migrate_repository:main
""",
test_suite = "py.test.cmdline.main",
)
Index: migrate/versioning/migrate_repository.py
===================================================================
--- migrate/versioning/migrate_repository.py (revision 444)
+++ migrate/versioning/migrate_repository.py (working copy)
@@ -4,7 +4,7 @@
def usage():
-
+
print '''Usage: %(prog)s repository-to-migrate
Upgrade your repository to the new flat format.
@@ -28,8 +28,8 @@
def deleteDirectory(dirpath):
print ' Deleting directory: %s' % dirpath
os.rmdir(dirpath)
-
+
def migrate_repository(repos):
print 'Migrating repository at: %s to new format' % repos
versions = '%s/versions' % repos
@@ -42,11 +42,11 @@
files = os.listdir(origdir)
files.sort()
for file in files:
-
+
# Delete compiled Python files.
if file.endswith('.pyc') or file.endswith('.pyo'):
deleteFile('%s/%s' % (origdir, file))
-
+
# Delete empty __init__.py files.
origfile = '%s/__init__.py' % origdir
if os.path.exists(origfile) and len(open(origfile).read()) == 0:
@@ -69,10 +69,12 @@
# Try to remove directory. Will fail if it's not empty.
deleteDirectory(origdir)
-
-if __name__ == '__main__':
+
+def main():
if len(sys.argv) != 2:
usage()
migrate_repository(sys.argv[1])
+if __name__ == '__main__':
+ main()