You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
8 months ago
|
import argparse
|
||
|
import os
|
||
|
import secrets
|
||
|
import sys
|
||
|
import textwrap
|
||
|
|
||
|
from passlib.hash import pbkdf2_sha512
|
||
|
|
||
|
from . import Command
|
||
|
from odoo.tools import config
|
||
|
|
||
|
|
||
|
class GenProxyToken(Command):
|
||
|
def __init__(self):
|
||
|
self.command_name = "genproxytoken"
|
||
|
|
||
|
def generate_token(self, length=16):
|
||
|
token = secrets.token_hex(int(length / 2))
|
||
|
split_size = int(length / 4)
|
||
|
return '-'.join(textwrap.wrap(token, split_size))
|
||
|
|
||
|
def run(self, cmdargs):
|
||
|
parser = argparse.ArgumentParser(
|
||
|
prog="%s %s" % (sys.argv[0].split(os.path.sep)[-1], self.command_name),
|
||
|
description="Generate and (re)set proxy access token in config file"
|
||
|
)
|
||
|
parser.add_argument('-c', '--config', type=str, help="Specify an alternate config file")
|
||
|
parser.add_argument('--token-length', type=int, help="Token Length", default=16)
|
||
|
args, _ = parser.parse_known_args()
|
||
|
if args.config:
|
||
|
config.rcfile = args.config
|
||
|
token = self.generate_token(length=args.token_length)
|
||
|
config['proxy_access_token'] = pbkdf2_sha512.hash(token)
|
||
|
config.save()
|
||
|
sys.stdout.write(f'{token}\n')
|