StackStorm /
st2
| Conditions | 15 |
| Total Lines | 95 |
| Lines | 0 |
| Ratio | 0 % |
| Changes | 1 | ||
| Bugs | 0 | Features | 0 |
Small methods make your code easier to understand, in particular if combined with a good name. Besides, if your method is small, finding a good name is usually much easier.
For example, if you find yourself adding comments to a method's body, this is usually a good sign to extract the commented part to a new method, and use the comment as a starting point when coming up with a good name for this new method.
Commonly applied refactorings include:
If many parameters/temporary variables are present:
Complex classes like TokenController._handle_standalone_auth() often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.
Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.
| 1 | # Licensed to the StackStorm, Inc ('StackStorm') under one or more |
||
| 92 | def _handle_standalone_auth(self, request, **kwargs): |
||
| 93 | authorization = pecan.request.authorization |
||
| 94 | |||
| 95 | auth_backend = self._auth_backend.__class__.__name__ |
||
| 96 | remote_addr = pecan.request.remote_addr |
||
| 97 | extra = {'auth_backend': auth_backend, 'remote_addr': remote_addr} |
||
| 98 | |||
| 99 | if not authorization: |
||
| 100 | LOG.audit('Authorization header not provided', extra=extra) |
||
| 101 | self._abort_request() |
||
| 102 | return |
||
| 103 | |||
| 104 | auth_type, auth_value = authorization |
||
| 105 | if auth_type.lower() not in ['basic']: |
||
| 106 | extra['auth_type'] = auth_type |
||
| 107 | LOG.audit('Unsupported authorization type: %s' % (auth_type), extra=extra) |
||
| 108 | self._abort_request() |
||
| 109 | return |
||
| 110 | |||
| 111 | try: |
||
| 112 | auth_value = base64.b64decode(auth_value) |
||
| 113 | except Exception: |
||
| 114 | LOG.audit('Invalid authorization header', extra=extra) |
||
| 115 | self._abort_request() |
||
| 116 | return |
||
| 117 | |||
| 118 | split = auth_value.split(':') |
||
| 119 | if len(split) != 2: |
||
| 120 | LOG.audit('Invalid authorization header', extra=extra) |
||
| 121 | self._abort_request() |
||
| 122 | return |
||
| 123 | |||
| 124 | username, password = split |
||
| 125 | result = self._auth_backend |
||
| 126 | |||
| 127 | result = self._auth_backend.authenticate(username=username, password=password) |
||
| 128 | if result is True: |
||
| 129 | LOG.audit(request.body) |
||
| 130 | ttl = getattr(request, 'ttl', None) |
||
| 131 | impersonate_user = getattr(request.body, 'user', None) |
||
| 132 | |||
| 133 | if impersonate_user is not None: |
||
| 134 | # check this is a service account |
||
| 135 | if not User.get_by_name(username).is_service: |
||
| 136 | message = "Current user is not a service and cannot " \ |
||
| 137 | "request impersonated tokens" |
||
| 138 | self._abort_request(status_code=http_client.BAD_REQUEST, |
||
| 139 | message=message) |
||
| 140 | return |
||
| 141 | username = impersonate_user |
||
| 142 | else: |
||
| 143 | impersonate_user = getattr(request, 'impersonate_user', None) |
||
| 144 | nickname_origin = getattr(request, 'nickname_origin', None) |
||
| 145 | if impersonate_user is not None: |
||
| 146 | try: |
||
| 147 | # check this is a service account |
||
| 148 | if not User.get_by_name(username).is_service: |
||
| 149 | raise NotServiceUserError() |
||
| 150 | username = User.get_by_nickname(impersonate_user, |
||
| 151 | nickname_origin).name |
||
| 152 | except NotServiceUserError: |
||
| 153 | message = "Current user is not a service and cannot " \ |
||
| 154 | "request impersonated tokens" |
||
| 155 | self._abort_request(status_code=http_client.BAD_REQUEST, |
||
| 156 | message=message) |
||
| 157 | return |
||
| 158 | except UserNotFoundError: |
||
| 159 | message = "Could not locate user %s@%s" % \ |
||
| 160 | (impersonate_user, nickname_origin) |
||
| 161 | self._abort_request(status_code=http_client.BAD_REQUEST, |
||
| 162 | message=message) |
||
| 163 | return |
||
| 164 | except NoNicknameOriginProvidedError: |
||
| 165 | message = "Nickname origin is not provided for nickname '%s'" % \ |
||
| 166 | impersonate_user |
||
| 167 | self._abort_request(status_code=http_client.BAD_REQUEST, |
||
| 168 | message=message) |
||
| 169 | return |
||
| 170 | except AmbiguousUserError: |
||
| 171 | message = "%s@%s matched more than one username" % \ |
||
| 172 | (impersonate_user, nickname_origin) |
||
| 173 | self._abort_request(status_code=http_client.BAD_REQUEST, |
||
| 174 | message=message) |
||
| 175 | return |
||
| 176 | try: |
||
| 177 | token = self._create_token_for_user( |
||
| 178 | username=username, ttl=ttl) |
||
| 179 | return self._process_successful_response(token=token) |
||
| 180 | except TTLTooLargeException as e: |
||
| 181 | self._abort_request(status_code=http_client.BAD_REQUEST, |
||
| 182 | message=e.message) |
||
| 183 | return |
||
| 184 | |||
| 185 | LOG.audit('Invalid credentials provided', extra=extra) |
||
| 186 | self._abort_request() |
||
| 187 | |||
| 200 |
