add path argument for both commands

Author: freenoth

.gitignore

@@ -1,6 +1,6 @@
 # Created by .ignore support plugin (hsz.mobi)
 .idea
 manage.py
-django-auto-rollback
+django-rollback
 __pycache__
 db.sqlite3

README.md

@@ -3,7 +3,7 @@ This package allow to easy manage apps migrations based on GIT repository (it us
 So you can return to any previous state that is saved in DB by one command.
 
 ## Version
-Current version is `0.2.0`
+Current version is `0.2.1`
 
 It works with:
 - django == 1.11
@@ -14,7 +14,7 @@ It works with:
 
 First you need to install package with pip:
 ```bash
-pip install git+https://github.com/freenoth/[email protected].0
+pip install git+https://github.com/freenoth/[email protected].1
 ```
 
 Then install it to your `INSTALLED_APPS`
@@ -28,12 +28,23 @@ INSTALLED_APPS = [
 You are also should run `./manage.py migrate` before using additional management commands.
 
 ## Using
-There are two commands to manage migrations state:
+There are two commands to manage migrations state.
+
+For both commands you can pass the `PATH` argument to specify path to git repository directory (local). Default path is current dir : `'.'`.
 
 ### Saving current state
 ```bash
 ./manage.py save_migrations_state
 ```
+Help message below:
+```bash
+usage: manage.py save_migrations_state [-p PATH]
+
+Save migrations state for current commit.
+
+optional arguments:
+  -p PATH, --path PATH  Git repository path.
+```
 This command used to save apps migrations state of current commit to DB (it create new or update existing state).
 
 Successful output example below:
@@ -48,7 +59,7 @@ Successfully created for commit "84e47461a95fa325d9e933bbe8cca8c52bbea203".
 ```
 Help message below:
 ```bash
-usage: manage.py rollback_migrations [-t TAG] [-c COMMIT] [--fake]
+usage: manage.py rollback_migrations [-t TAG] [-c COMMIT] [-p PATH] [--fake]
 
 Rollback migrations state of all django apps to chosen tag or commit if
 previously saved.
@@ -57,6 +68,7 @@ optional arguments:
   -t TAG, --tag TAG     Git tag to which to rollback migrations.
   -c COMMIT, --commit COMMIT
                         Git commit hash to which to rollback migrations.
+  -p PATH, --path PATH  Git repository path.
   --fake                It allow to only print info about processed actions
                         without execution (no changes for DB).
 

django_rollback/management/commands/rollback_migrations.py

@@ -7,6 +7,7 @@
 
 from django_rollback.models import AppsState
 
+DEFAULT_REPO_PATH = '.'
 COMMIT_MAX_LENGTH = 40
 MIGRATE_COMMAND = 'migrate'
 
@@ -19,13 +20,16 @@ class Command(BaseCommand):
     def add_arguments(self, parser):
         parser.add_argument('-t', '--tag', type=str, help='Git tag to which to rollback migrations.')
         parser.add_argument('-c', '--commit', type=str, help='Git commit hash to which to rollback migrations.')
+        parser.add_argument('-p', '--path', type=str, help='Git repository path.')
         parser.add_argument('--fake', action='store_true',
                             help='It allow to only print info about processed actions without execution '
                                  '(no changes for DB).')
 
     def handle(self, *args, **options):
-        other_commit = self.get_commit_from_options(options)
-        current_commit = self.get_current_commit()
+        repo_path = options.get('path', DEFAULT_REPO_PATH)
+
+        other_commit = self.get_commit_from_options(options, path=repo_path)
+        current_commit = self.get_current_commit(path=repo_path)
 
         current_data = self.get_migrations_data_from_commit(current_commit)
         other_data = self.get_migrations_data_from_commit(other_commit)
@@ -36,7 +40,7 @@ def handle(self, *args, **options):
 
         self.stdout.write(self.style.SUCCESS('Rollback successfully finished.'))
 
-    def get_commit_from_options(self, options):
+    def get_commit_from_options(self, options, path):
         if not options['tag'] and not options['commit']:
             raise CommandError('Tag or commit should be described by -t or -c arguments.')
 
@@ -48,7 +52,7 @@ def get_commit_from_options(self, options):
 
         tag = options['tag']
         try:
-            repo = git.Repo('.')
+            repo = git.Repo(path)
             if tag not in repo.tags:
                 raise CommandError(f'Can not find tag `{tag}` in git repository.')
 
@@ -61,9 +65,9 @@ def get_commit_from_options(self, options):
             self.stdout.write(self.style.ERROR(f'WARNING: an error occurred while working with git repo!'))
             raise CommandError(err)
 
-    def get_current_commit(self):
+    def get_current_commit(self, path):
         try:
-            repo = git.Repo('.')
+            repo = git.Repo(path)
             return repo.head.commit.hexsha
 
         except ValueError as err:

django_rollback/management/commands/save_migrations_state.py

@@ -7,12 +7,17 @@
 from django_rollback.models import AppsState
 from django_rollback.sql import MIGRATIONS_STATE_SQL
 
+DEFAULT_REPO_PATH = '.'
+
 
 class Command(BaseCommand):
     help = 'Save migrations state for current commit.'
 
+    def add_arguments(self, parser):
+        parser.add_argument('-p', '--path', type=str, help='Git repository path.')
+
     def handle(self, *args, **options):
-        commit = self.get_current_commit()
+        commit = self.get_current_commit(path=options.get('path', DEFAULT_REPO_PATH))
         state_data = self.get_current_migrations_state()
 
         obj, created = AppsState.objects.update_or_create(commit=commit,
@@ -22,9 +27,9 @@ def handle(self, *args, **options):
                                              f'Successfully {"created" if created else "updated"} '
                                              f'for commit "{commit}".\n'))
 
-    def get_current_commit(self):
+    def get_current_commit(self, path):
         try:
-            repo = git.Repo('.')
+            repo = git.Repo(path)
             return repo.head.commit.hexsha
         except ValueError as err:
             self.stdout.write(self.style.ERROR(f'WARNING: an error occurred while working with git repo!'))

setup.py

@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name='django_rollback',
-    version='0.2.0',
+    version='0.2.1',
     author='Aleksey Yakovlev',
     author_email='[email protected]',
     description='Package to automate django rollback',