layerindex-web/layerindex/tools/site_name.py
Tim Orling 4e857f68f4 layerindex/*: make all shebangs python3
Several scripts still had /usr/bin/env python

Signed-off-by: Tim Orling <tim.orling@konsulko.com>
2022-01-13 20:58:28 -08:00

74 lines
1.6 KiB
Python
Executable File

#!/usr/bin/env python3
# Updates site name in Django database
#
# Copyright (C) 2019 Intel Corporation
# Author: Paul Eggleton <paul.eggleton@linux.intel.com>
#
# Licensed under the MIT license, see COPYING.MIT for details
#
# SPDX-License-Identifier: MIT
import sys
import os
import argparse
import logging
sys.path.insert(0, os.path.realpath(os.path.join(os.path.dirname(__file__), '..')))
import utils
logger = utils.logger_create('LayerIndexComparisonUpdate')
class DryRunRollbackException(Exception):
pass
def set_site_name(args):
utils.setup_django()
from django.contrib.sites.models import Site
site = Site.objects.get_current()
if not args.domain:
if not site:
print('No site object currently defined')
return 1
else:
print('%s\t%s' % (site.domain, site.name))
return 0
if not site:
site = Site()
if args.domain:
site.domain = args.domain
if args.name:
site.name = args.name
site.save()
return 0
def main():
parser = argparse.ArgumentParser(description="Set site name tool",
epilog="With no arguments, site domain/name will be printed.")
parser.add_argument('domain', nargs='?', help='Site domain to set')
parser.add_argument('name', nargs='?', help='Site descriptive name to set')
args = parser.parse_args()
ret = set_site_name(args)
return ret
if __name__ == "__main__":
try:
ret = main()
except Exception:
ret = 1
import traceback
traceback.print_exc()
sys.exit(ret)