def handle(self, *args, **options): if options.get('server_ip') and options.get('user') and options.get('password'): config_data = options elif options.get('config'): config_data = wr.read_server_config(options.get('config')) else: raise CommandError("Server credentials are wrong or missing") wr.set_server_config(config_data) wr.upgrade_to_superuser(DB_USER) wr.create_hstore_extenstion() self.output("HStore extension successfully created.", style="success")