| OLD | NEW |
| 1 # coding: utf-8 | 1 # coding: utf-8 |
| 2 # Copyright 2015 The LUCI Authors. All rights reserved. | 2 # Copyright 2015 The LUCI Authors. All rights reserved. |
| 3 # Use of this source code is governed under the Apache License, Version 2.0 | 3 # Use of this source code is governed under the Apache License, Version 2.0 |
| 4 # that can be found in the LICENSE file. | 4 # that can be found in the LICENSE file. |
| 5 | 5 |
| 6 """Packing and unpacking of ndb.Key. | 6 """Packing and unpacking of ndb.Key. |
| 7 | 7 |
| 8 End users are only given packed keys, which permits to not expose internal | 8 End users are only given packed keys, which permits to not expose internal |
| 9 schema details to the user. | 9 schema details to the user. |
| 10 """ | 10 """ |
| (...skipping 60 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 71 """Returns the PerformanceStats ndb.Key for this TaskRunResult.key. | 71 """Returns the PerformanceStats ndb.Key for this TaskRunResult.key. |
| 72 """ | 72 """ |
| 73 assert run_result_key.kind() == 'TaskRunResult', run_result_key | 73 assert run_result_key.kind() == 'TaskRunResult', run_result_key |
| 74 return ndb.Key('PerformanceStats', 1, parent=run_result_key) | 74 return ndb.Key('PerformanceStats', 1, parent=run_result_key) |
| 75 | 75 |
| 76 | 76 |
| 77 ### Packing and unpacking. | 77 ### Packing and unpacking. |
| 78 | 78 |
| 79 | 79 |
| 80 def get_request_and_result_keys(task_id): | 80 def get_request_and_result_keys(task_id): |
| 81 """Provides the key and TaskRequest corresponding to a task ID. | 81 """Provides request and result keys corresponding to a task ID. |
| 82 | 82 |
| 83 Returns: | 83 Returns: |
| 84 tuple(request_key, result_key): ndb.Key that yield TaskRequest and either | 84 tuple(request_key, result_key): ndb.Key that yield TaskRequest and either |
| 85 (TaskRunResult or TaskResultSummay). | 85 (TaskRunResult or TaskResultSummary). |
| 86 """ | 86 """ |
| 87 try: | 87 try: |
| 88 key = unpack_result_summary_key(task_id) | 88 key = unpack_result_summary_key(task_id) |
| 89 request_key = result_summary_key_to_request_key(key) | 89 request_key = result_summary_key_to_request_key(key) |
| 90 except ValueError: | 90 except ValueError: |
| 91 key = unpack_run_result_key(task_id) | 91 key = unpack_run_result_key(task_id) |
| 92 request_key = result_summary_key_to_request_key( | 92 request_key = result_summary_key_to_request_key( |
| 93 run_result_key_to_result_summary_key(key)) | 93 run_result_key_to_result_summary_key(key)) |
| 94 return request_key, key | 94 return request_key, key |
| 95 | 95 |
| (...skipping 68 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 164 """Returns the TaskRunResult ndb.Key from a packed key. | 164 """Returns the TaskRunResult ndb.Key from a packed key. |
| 165 | 165 |
| 166 The expected format of |packed_key| is %x. | 166 The expected format of |packed_key| is %x. |
| 167 """ | 167 """ |
| 168 request_key = unpack_request_key(packed_key[:-1]) | 168 request_key = unpack_request_key(packed_key[:-1]) |
| 169 run_id = int(packed_key[-1], 16) | 169 run_id = int(packed_key[-1], 16) |
| 170 if not run_id: | 170 if not run_id: |
| 171 raise ValueError('Can\'t reference to the overall task result.') | 171 raise ValueError('Can\'t reference to the overall task result.') |
| 172 result_summary_key = request_key_to_result_summary_key(request_key) | 172 result_summary_key = request_key_to_result_summary_key(request_key) |
| 173 return result_summary_key_to_run_result_key(result_summary_key, run_id) | 173 return result_summary_key_to_run_result_key(result_summary_key, run_id) |
| OLD | NEW |