From dcfed84414d3340c70380c65e88571f493e7a762 Mon Sep 17 00:00:00 2001
From: Davis Raymond Muro <davisraymondmuro@outlook.com>
Date: Thu, 16 Jul 2020 14:48:52 +0300
Subject: [PATCH] Add "increase_odk_token_lifetime" management command

---
 .../commands/increase_odk_token_lifetime.py   | 50 +++++++++++++++++++
 .../test_increase_odk_token_lifetime.py       | 30 +++++++++++
 2 files changed, 80 insertions(+)
 create mode 100644 onadata/apps/api/management/commands/increase_odk_token_lifetime.py
 create mode 100644 onadata/apps/api/tests/management/commands/test_increase_odk_token_lifetime.py

diff --git a/onadata/apps/api/management/commands/increase_odk_token_lifetime.py b/onadata/apps/api/management/commands/increase_odk_token_lifetime.py
new file mode 100644
index 0000000000..e98caeb687
--- /dev/null
+++ b/onadata/apps/api/management/commands/increase_odk_token_lifetime.py
@@ -0,0 +1,50 @@
+from datetime import timedelta
+
+from django.core.management.base import BaseCommand, CommandError
+from django.utils.translation import gettext as _
+
+from onadata.apps.api.models.odk_token import ODKToken
+
+
+def increase_odk_token_lifetime(days: int, username: str):
+    qs = ODKToken.objects.filter(
+            user__username=username, status=ODKToken.ACTIVE)
+    if qs.count() < 1:
+        return False
+
+    token = qs.first()
+    updated_expiry_date = token.expires + timedelta(days=days)
+    token.expires = updated_expiry_date.astimezone(token.expires.tzinfo)
+    token.save()
+    return True
+
+
+class Command(BaseCommand):
+    help = _("Increase ODK Token lifetime for a particular user.")
+
+    def add_arguments(self, parser):
+        parser.add_argument(
+            '--days',
+            '-d',
+            default=30,
+            dest='days',
+            help='Number of days to increase the token lifetime by.')
+        parser.add_argument(
+            '--username',
+            '-u',
+            dest='username',
+            help='The users username'
+        )
+
+    def handle(self, *args, **options):
+        username = options.get('username')
+        days = options.get('days')
+
+        if not username:
+            raise CommandError('No username provided.')
+
+        created = increase_odk_token_lifetime(days, username)
+        if not created:
+            raise CommandError(f'User {username} has no active ODK Token.')
+        self.stdout.write(
+            f'Increased the lifetime of ODK Token for user {username}')
diff --git a/onadata/apps/api/tests/management/commands/test_increase_odk_token_lifetime.py b/onadata/apps/api/tests/management/commands/test_increase_odk_token_lifetime.py
new file mode 100644
index 0000000000..3cd040d68a
--- /dev/null
+++ b/onadata/apps/api/tests/management/commands/test_increase_odk_token_lifetime.py
@@ -0,0 +1,30 @@
+from datetime import timedelta
+
+from django.contrib.auth.models import User
+from django.core.management import call_command
+from django.utils.six import StringIO
+
+from onadata.apps.main.tests.test_base import TestBase
+from onadata.apps.api.models.odk_token import ODKToken
+
+
+class IncreaseODKTokenLifetimeTest(TestBase):
+    def test_increase_odk_token_lifetime(self):
+        user = User.objects.create(
+            username='dave', email='dave@example.com')
+        token = ODKToken.objects.create(user=user)
+        expiry_date = token.expires
+
+        out = StringIO()
+        call_command(
+            'increase_odk_token_lifetime',
+            days=2,
+            username=user.username,
+            stdout=out
+        )
+
+        self.assertEqual(
+            'Increased the lifetime of ODK Token for user dave\n',
+            out.getvalue())
+        token.refresh_from_db()
+        self.assertEqual(expiry_date + timedelta(days=2), token.expires)