Add some QoL scripts for restoring a database and extending a user
license
This commit is contained in:
parent
6f4d50b033
commit
5e78d08c48
|
|
@ -0,0 +1,11 @@
|
||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
while getopts d:f: flag
|
||||||
|
do
|
||||||
|
case "${flag}" in
|
||||||
|
d) database=${OPTARG};;
|
||||||
|
f) filename=${OPTARG};;
|
||||||
|
esac
|
||||||
|
done
|
||||||
|
|
||||||
|
pg_restore --verbose --clean --no-acl --no-owner -d $database $filename
|
||||||
|
|
@ -0,0 +1,24 @@
|
||||||
|
from datetime import date, timedelta
|
||||||
|
from django.core.management import BaseCommand
|
||||||
|
|
||||||
|
from users.models import User
|
||||||
|
|
||||||
|
|
||||||
|
class Command(BaseCommand):
|
||||||
|
def add_arguments(self, parser):
|
||||||
|
parser.add_argument("username")
|
||||||
|
|
||||||
|
def handle(self, *args, **options):
|
||||||
|
username = options["username"]
|
||||||
|
try:
|
||||||
|
user = User.objects.get(username=username)
|
||||||
|
today = date.today()
|
||||||
|
delta = timedelta(180)
|
||||||
|
user.license_expiry_date = today + delta
|
||||||
|
user.save()
|
||||||
|
self.stdout.write(
|
||||||
|
f"new date set for user{user.username}: {user.license_expiry_date}"
|
||||||
|
)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
self.stdout.write(e)
|
||||||
Loading…
Reference in New Issue