OpenDroneMap-WebODM/app/management/commands/addnode.py

23 wiersze
918 B
Python

import os
from django.core.management.base import BaseCommand
from nodeodm.models import ProcessingNode
class Command(BaseCommand):
requires_system_checks = []
def add_arguments(self, parser):
parser.add_argument("host", type=str)
parser.add_argument("port", type=int)
parser.add_argument("--label", type=str, required=False, default="", help="Node label")
parser.add_argument("--token", type=str, required=False, default="", help="Node token")
super(Command, self).add_arguments(parser)
def handle(self, **options):
ProcessingNode.objects.update_or_create(hostname=options.get('host'),
defaults={
'hostname': options.get('host'),
'port': options.get('port'),
'label': options.get('label', ''),
'token': options.get('token', '')
})