forked from enviPath/enviPy
Fixes #12 Co-authored-by: Tim Lorsbach <tim@lorsba.ch> Reviewed-on: enviPath/enviPy#30
28 lines
748 B
Python
28 lines
748 B
Python
from django.core.management.base import BaseCommand
|
|
|
|
from epdb.logic import PackageManager
|
|
from epdb.models import *
|
|
|
|
|
|
class Command(BaseCommand):
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument(
|
|
'--data',
|
|
type=str,
|
|
help='Path of the Package to import.',
|
|
required=True,
|
|
)
|
|
parser.add_argument(
|
|
'--owner',
|
|
type=str,
|
|
help='Username of the desired Owner.',
|
|
required=True,
|
|
)
|
|
|
|
@transaction.atomic
|
|
def handle(self, *args, **options):
|
|
owner = User.objects.get(username=options['owner'])
|
|
package_data = json.load(open(options['data']))
|
|
PackageManager.import_package(package_data, owner)
|