Ejemplo n.º 1
0
    def post(self):

        if request.remote_addr != '127.0.0.1':
            return {
                'message': "Your are'nt authorized to access this route"
            }, 401

        data = parser.parse_args()

        if UserModel.find_by_name(data['name']):
            return {'message': 'User {} already exists'.format(data['name'])}

        new_user = UserModel(name=data['name'],
                             password=UserModel.generate_hash(
                                 data['password']))

        try:
            new_user.save_to_db()
            access_token = create_access_token(
                identity=data['name'],
                expires_delta=datetime.timedelta(hours=1))
            refresh_token = create_refresh_token(identity=data['name'])
            return {
                'message': 'User {} was created'.format(data['name']),
                'access_token': access_token,
                'refresh_token': refresh_token
            }
        except:
            return {'message': 'Something went wrong'}, 500
Ejemplo n.º 2
0
def register(user, password):
    userobj = UserModel.find_by_name(user)

    if not userobj:
        return 'User {} do not exists, create it via sshportal command line'.format(
            user)

    password = getpass('Please enter your password: '******'User {} updated'.format(user)
    except:
        return 'Something went wrong'