rucio 37.0.0rc1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of rucio might be problematic. Click here for more details.

Files changed (487) hide show
  1. rucio/__init__.py +17 -0
  2. rucio/alembicrevision.py +15 -0
  3. rucio/cli/__init__.py +14 -0
  4. rucio/cli/account.py +216 -0
  5. rucio/cli/bin_legacy/__init__.py +13 -0
  6. rucio/cli/bin_legacy/rucio.py +2825 -0
  7. rucio/cli/bin_legacy/rucio_admin.py +2500 -0
  8. rucio/cli/command.py +272 -0
  9. rucio/cli/config.py +72 -0
  10. rucio/cli/did.py +191 -0
  11. rucio/cli/download.py +128 -0
  12. rucio/cli/lifetime_exception.py +33 -0
  13. rucio/cli/replica.py +162 -0
  14. rucio/cli/rse.py +293 -0
  15. rucio/cli/rule.py +158 -0
  16. rucio/cli/scope.py +40 -0
  17. rucio/cli/subscription.py +73 -0
  18. rucio/cli/upload.py +60 -0
  19. rucio/cli/utils.py +226 -0
  20. rucio/client/__init__.py +15 -0
  21. rucio/client/accountclient.py +432 -0
  22. rucio/client/accountlimitclient.py +183 -0
  23. rucio/client/baseclient.py +983 -0
  24. rucio/client/client.py +120 -0
  25. rucio/client/configclient.py +126 -0
  26. rucio/client/credentialclient.py +59 -0
  27. rucio/client/didclient.py +868 -0
  28. rucio/client/diracclient.py +56 -0
  29. rucio/client/downloadclient.py +1783 -0
  30. rucio/client/exportclient.py +44 -0
  31. rucio/client/fileclient.py +50 -0
  32. rucio/client/importclient.py +42 -0
  33. rucio/client/lifetimeclient.py +90 -0
  34. rucio/client/lockclient.py +109 -0
  35. rucio/client/metaconventionsclient.py +140 -0
  36. rucio/client/pingclient.py +44 -0
  37. rucio/client/replicaclient.py +452 -0
  38. rucio/client/requestclient.py +125 -0
  39. rucio/client/richclient.py +317 -0
  40. rucio/client/rseclient.py +746 -0
  41. rucio/client/ruleclient.py +294 -0
  42. rucio/client/scopeclient.py +90 -0
  43. rucio/client/subscriptionclient.py +173 -0
  44. rucio/client/touchclient.py +82 -0
  45. rucio/client/uploadclient.py +969 -0
  46. rucio/common/__init__.py +13 -0
  47. rucio/common/bittorrent.py +234 -0
  48. rucio/common/cache.py +111 -0
  49. rucio/common/checksum.py +168 -0
  50. rucio/common/client.py +122 -0
  51. rucio/common/config.py +788 -0
  52. rucio/common/constants.py +217 -0
  53. rucio/common/constraints.py +17 -0
  54. rucio/common/didtype.py +237 -0
  55. rucio/common/dumper/__init__.py +342 -0
  56. rucio/common/dumper/consistency.py +497 -0
  57. rucio/common/dumper/data_models.py +362 -0
  58. rucio/common/dumper/path_parsing.py +75 -0
  59. rucio/common/exception.py +1208 -0
  60. rucio/common/extra.py +31 -0
  61. rucio/common/logging.py +420 -0
  62. rucio/common/pcache.py +1409 -0
  63. rucio/common/plugins.py +185 -0
  64. rucio/common/policy.py +93 -0
  65. rucio/common/schema/__init__.py +200 -0
  66. rucio/common/schema/generic.py +416 -0
  67. rucio/common/schema/generic_multi_vo.py +395 -0
  68. rucio/common/stomp_utils.py +423 -0
  69. rucio/common/stopwatch.py +55 -0
  70. rucio/common/test_rucio_server.py +154 -0
  71. rucio/common/types.py +483 -0
  72. rucio/common/utils.py +1688 -0
  73. rucio/core/__init__.py +13 -0
  74. rucio/core/account.py +496 -0
  75. rucio/core/account_counter.py +236 -0
  76. rucio/core/account_limit.py +425 -0
  77. rucio/core/authentication.py +620 -0
  78. rucio/core/config.py +437 -0
  79. rucio/core/credential.py +224 -0
  80. rucio/core/did.py +3004 -0
  81. rucio/core/did_meta_plugins/__init__.py +252 -0
  82. rucio/core/did_meta_plugins/did_column_meta.py +331 -0
  83. rucio/core/did_meta_plugins/did_meta_plugin_interface.py +165 -0
  84. rucio/core/did_meta_plugins/elasticsearch_meta.py +407 -0
  85. rucio/core/did_meta_plugins/filter_engine.py +672 -0
  86. rucio/core/did_meta_plugins/json_meta.py +240 -0
  87. rucio/core/did_meta_plugins/mongo_meta.py +229 -0
  88. rucio/core/did_meta_plugins/postgres_meta.py +352 -0
  89. rucio/core/dirac.py +237 -0
  90. rucio/core/distance.py +187 -0
  91. rucio/core/exporter.py +59 -0
  92. rucio/core/heartbeat.py +363 -0
  93. rucio/core/identity.py +301 -0
  94. rucio/core/importer.py +260 -0
  95. rucio/core/lifetime_exception.py +377 -0
  96. rucio/core/lock.py +577 -0
  97. rucio/core/message.py +288 -0
  98. rucio/core/meta_conventions.py +203 -0
  99. rucio/core/monitor.py +448 -0
  100. rucio/core/naming_convention.py +195 -0
  101. rucio/core/nongrid_trace.py +136 -0
  102. rucio/core/oidc.py +1463 -0
  103. rucio/core/permission/__init__.py +161 -0
  104. rucio/core/permission/generic.py +1124 -0
  105. rucio/core/permission/generic_multi_vo.py +1144 -0
  106. rucio/core/quarantined_replica.py +224 -0
  107. rucio/core/replica.py +4483 -0
  108. rucio/core/replica_sorter.py +362 -0
  109. rucio/core/request.py +3091 -0
  110. rucio/core/rse.py +2079 -0
  111. rucio/core/rse_counter.py +185 -0
  112. rucio/core/rse_expression_parser.py +459 -0
  113. rucio/core/rse_selector.py +304 -0
  114. rucio/core/rule.py +4484 -0
  115. rucio/core/rule_grouping.py +1620 -0
  116. rucio/core/scope.py +181 -0
  117. rucio/core/subscription.py +362 -0
  118. rucio/core/topology.py +490 -0
  119. rucio/core/trace.py +375 -0
  120. rucio/core/transfer.py +1531 -0
  121. rucio/core/vo.py +169 -0
  122. rucio/core/volatile_replica.py +151 -0
  123. rucio/daemons/__init__.py +13 -0
  124. rucio/daemons/abacus/__init__.py +13 -0
  125. rucio/daemons/abacus/account.py +116 -0
  126. rucio/daemons/abacus/collection_replica.py +124 -0
  127. rucio/daemons/abacus/rse.py +117 -0
  128. rucio/daemons/atropos/__init__.py +13 -0
  129. rucio/daemons/atropos/atropos.py +242 -0
  130. rucio/daemons/auditor/__init__.py +289 -0
  131. rucio/daemons/auditor/hdfs.py +97 -0
  132. rucio/daemons/auditor/srmdumps.py +355 -0
  133. rucio/daemons/automatix/__init__.py +13 -0
  134. rucio/daemons/automatix/automatix.py +304 -0
  135. rucio/daemons/badreplicas/__init__.py +13 -0
  136. rucio/daemons/badreplicas/minos.py +322 -0
  137. rucio/daemons/badreplicas/minos_temporary_expiration.py +171 -0
  138. rucio/daemons/badreplicas/necromancer.py +196 -0
  139. rucio/daemons/bb8/__init__.py +13 -0
  140. rucio/daemons/bb8/bb8.py +353 -0
  141. rucio/daemons/bb8/common.py +759 -0
  142. rucio/daemons/bb8/nuclei_background_rebalance.py +153 -0
  143. rucio/daemons/bb8/t2_background_rebalance.py +153 -0
  144. rucio/daemons/cache/__init__.py +13 -0
  145. rucio/daemons/cache/consumer.py +133 -0
  146. rucio/daemons/common.py +405 -0
  147. rucio/daemons/conveyor/__init__.py +13 -0
  148. rucio/daemons/conveyor/common.py +562 -0
  149. rucio/daemons/conveyor/finisher.py +529 -0
  150. rucio/daemons/conveyor/poller.py +394 -0
  151. rucio/daemons/conveyor/preparer.py +205 -0
  152. rucio/daemons/conveyor/receiver.py +179 -0
  153. rucio/daemons/conveyor/stager.py +133 -0
  154. rucio/daemons/conveyor/submitter.py +403 -0
  155. rucio/daemons/conveyor/throttler.py +532 -0
  156. rucio/daemons/follower/__init__.py +13 -0
  157. rucio/daemons/follower/follower.py +101 -0
  158. rucio/daemons/hermes/__init__.py +13 -0
  159. rucio/daemons/hermes/hermes.py +534 -0
  160. rucio/daemons/judge/__init__.py +13 -0
  161. rucio/daemons/judge/cleaner.py +159 -0
  162. rucio/daemons/judge/evaluator.py +185 -0
  163. rucio/daemons/judge/injector.py +162 -0
  164. rucio/daemons/judge/repairer.py +154 -0
  165. rucio/daemons/oauthmanager/__init__.py +13 -0
  166. rucio/daemons/oauthmanager/oauthmanager.py +198 -0
  167. rucio/daemons/reaper/__init__.py +13 -0
  168. rucio/daemons/reaper/dark_reaper.py +282 -0
  169. rucio/daemons/reaper/reaper.py +739 -0
  170. rucio/daemons/replicarecoverer/__init__.py +13 -0
  171. rucio/daemons/replicarecoverer/suspicious_replica_recoverer.py +626 -0
  172. rucio/daemons/rsedecommissioner/__init__.py +13 -0
  173. rucio/daemons/rsedecommissioner/config.py +81 -0
  174. rucio/daemons/rsedecommissioner/profiles/__init__.py +24 -0
  175. rucio/daemons/rsedecommissioner/profiles/atlas.py +60 -0
  176. rucio/daemons/rsedecommissioner/profiles/generic.py +452 -0
  177. rucio/daemons/rsedecommissioner/profiles/types.py +93 -0
  178. rucio/daemons/rsedecommissioner/rse_decommissioner.py +280 -0
  179. rucio/daemons/storage/__init__.py +13 -0
  180. rucio/daemons/storage/consistency/__init__.py +13 -0
  181. rucio/daemons/storage/consistency/actions.py +848 -0
  182. rucio/daemons/tracer/__init__.py +13 -0
  183. rucio/daemons/tracer/kronos.py +511 -0
  184. rucio/daemons/transmogrifier/__init__.py +13 -0
  185. rucio/daemons/transmogrifier/transmogrifier.py +762 -0
  186. rucio/daemons/undertaker/__init__.py +13 -0
  187. rucio/daemons/undertaker/undertaker.py +137 -0
  188. rucio/db/__init__.py +13 -0
  189. rucio/db/sqla/__init__.py +52 -0
  190. rucio/db/sqla/constants.py +206 -0
  191. rucio/db/sqla/migrate_repo/__init__.py +13 -0
  192. rucio/db/sqla/migrate_repo/env.py +110 -0
  193. rucio/db/sqla/migrate_repo/versions/01eaf73ab656_add_new_rule_notification_state_progress.py +70 -0
  194. rucio/db/sqla/migrate_repo/versions/0437a40dbfd1_add_eol_at_in_rules.py +47 -0
  195. rucio/db/sqla/migrate_repo/versions/0f1adb7a599a_create_transfer_hops_table.py +59 -0
  196. rucio/db/sqla/migrate_repo/versions/102efcf145f4_added_stuck_at_column_to_rules.py +43 -0
  197. rucio/db/sqla/migrate_repo/versions/13d4f70c66a9_introduce_transfer_limits.py +91 -0
  198. rucio/db/sqla/migrate_repo/versions/140fef722e91_cleanup_distances_table.py +76 -0
  199. rucio/db/sqla/migrate_repo/versions/14ec5aeb64cf_add_request_external_host.py +43 -0
  200. rucio/db/sqla/migrate_repo/versions/156fb5b5a14_add_request_type_to_requests_idx.py +50 -0
  201. rucio/db/sqla/migrate_repo/versions/1677d4d803c8_split_rse_availability_into_multiple.py +68 -0
  202. rucio/db/sqla/migrate_repo/versions/16a0aca82e12_create_index_on_table_replicas_path.py +40 -0
  203. rucio/db/sqla/migrate_repo/versions/1803333ac20f_adding_provenance_and_phys_group.py +45 -0
  204. rucio/db/sqla/migrate_repo/versions/1a29d6a9504c_add_didtype_chck_to_requests.py +60 -0
  205. rucio/db/sqla/migrate_repo/versions/1a80adff031a_create_index_on_rules_hist_recent.py +40 -0
  206. rucio/db/sqla/migrate_repo/versions/1c45d9730ca6_increase_identity_length.py +140 -0
  207. rucio/db/sqla/migrate_repo/versions/1d1215494e95_add_quarantined_replicas_table.py +73 -0
  208. rucio/db/sqla/migrate_repo/versions/1d96f484df21_asynchronous_rules_and_rule_approval.py +74 -0
  209. rucio/db/sqla/migrate_repo/versions/1f46c5f240ac_add_bytes_column_to_bad_replicas.py +43 -0
  210. rucio/db/sqla/migrate_repo/versions/1fc15ab60d43_add_message_history_table.py +50 -0
  211. rucio/db/sqla/migrate_repo/versions/2190e703eb6e_move_rse_settings_to_rse_attributes.py +134 -0
  212. rucio/db/sqla/migrate_repo/versions/21d6b9dc9961_add_mismatch_scheme_state_to_requests.py +64 -0
  213. rucio/db/sqla/migrate_repo/versions/22cf51430c78_add_availability_column_to_table_rses.py +39 -0
  214. rucio/db/sqla/migrate_repo/versions/22d887e4ec0a_create_sources_table.py +64 -0
  215. rucio/db/sqla/migrate_repo/versions/25821a8a45a3_remove_unique_constraint_on_requests.py +51 -0
  216. rucio/db/sqla/migrate_repo/versions/25fc855625cf_added_unique_constraint_to_rules.py +41 -0
  217. rucio/db/sqla/migrate_repo/versions/269fee20dee9_add_repair_cnt_to_locks.py +43 -0
  218. rucio/db/sqla/migrate_repo/versions/271a46ea6244_add_ignore_availability_column_to_rules.py +44 -0
  219. rucio/db/sqla/migrate_repo/versions/277b5fbb41d3_switch_heartbeats_executable.py +53 -0
  220. rucio/db/sqla/migrate_repo/versions/27e3a68927fb_remove_replicas_tombstone_and_replicas_.py +38 -0
  221. rucio/db/sqla/migrate_repo/versions/2854cd9e168_added_rule_id_column.py +47 -0
  222. rucio/db/sqla/migrate_repo/versions/295289b5a800_processed_by_and__at_in_requests.py +45 -0
  223. rucio/db/sqla/migrate_repo/versions/2962ece31cf4_add_nbaccesses_column_in_the_did_table.py +45 -0
  224. rucio/db/sqla/migrate_repo/versions/2af3291ec4c_added_replicas_history_table.py +57 -0
  225. rucio/db/sqla/migrate_repo/versions/2b69addda658_add_columns_for_third_party_copy_read_.py +45 -0
  226. rucio/db/sqla/migrate_repo/versions/2b8e7bcb4783_add_config_table.py +69 -0
  227. rucio/db/sqla/migrate_repo/versions/2ba5229cb54c_add_submitted_at_to_requests_table.py +43 -0
  228. rucio/db/sqla/migrate_repo/versions/2cbee484dcf9_added_column_volume_to_rse_transfer_.py +42 -0
  229. rucio/db/sqla/migrate_repo/versions/2edee4a83846_add_source_to_requests_and_requests_.py +47 -0
  230. rucio/db/sqla/migrate_repo/versions/2eef46be23d4_change_tokens_pk.py +46 -0
  231. rucio/db/sqla/migrate_repo/versions/2f648fc909f3_index_in_rule_history_on_scope_name.py +40 -0
  232. rucio/db/sqla/migrate_repo/versions/3082b8cef557_add_naming_convention_table_and_closed_.py +67 -0
  233. rucio/db/sqla/migrate_repo/versions/30d5206e9cad_increase_oauthrequest_redirect_msg_.py +37 -0
  234. rucio/db/sqla/migrate_repo/versions/30fa38b6434e_add_index_on_service_column_in_the_message_table.py +44 -0
  235. rucio/db/sqla/migrate_repo/versions/3152492b110b_added_staging_area_column.py +77 -0
  236. rucio/db/sqla/migrate_repo/versions/32c7d2783f7e_create_bad_replicas_table.py +60 -0
  237. rucio/db/sqla/migrate_repo/versions/3345511706b8_replicas_table_pk_definition_is_in_.py +72 -0
  238. rucio/db/sqla/migrate_repo/versions/35ef10d1e11b_change_index_on_table_requests.py +42 -0
  239. rucio/db/sqla/migrate_repo/versions/379a19b5332d_create_rse_limits_table.py +65 -0
  240. rucio/db/sqla/migrate_repo/versions/384b96aa0f60_created_rule_history_tables.py +133 -0
  241. rucio/db/sqla/migrate_repo/versions/3ac1660a1a72_extend_distance_table.py +55 -0
  242. rucio/db/sqla/migrate_repo/versions/3ad36e2268b0_create_collection_replicas_updates_table.py +76 -0
  243. rucio/db/sqla/migrate_repo/versions/3c9df354071b_extend_waiting_request_state.py +60 -0
  244. rucio/db/sqla/migrate_repo/versions/3d9813fab443_add_a_new_state_lost_in_badfilesstatus.py +44 -0
  245. rucio/db/sqla/migrate_repo/versions/40ad39ce3160_add_transferred_at_to_requests_table.py +43 -0
  246. rucio/db/sqla/migrate_repo/versions/4207be2fd914_add_notification_column_to_rules.py +64 -0
  247. rucio/db/sqla/migrate_repo/versions/42db2617c364_create_index_on_requests_external_id.py +40 -0
  248. rucio/db/sqla/migrate_repo/versions/436827b13f82_added_column_activity_to_table_requests.py +43 -0
  249. rucio/db/sqla/migrate_repo/versions/44278720f774_update_requests_typ_sta_upd_idx_index.py +44 -0
  250. rucio/db/sqla/migrate_repo/versions/45378a1e76a8_create_collection_replica_table.py +78 -0
  251. rucio/db/sqla/migrate_repo/versions/469d262be19_removing_created_at_index.py +41 -0
  252. rucio/db/sqla/migrate_repo/versions/4783c1f49cb4_create_distance_table.py +59 -0
  253. rucio/db/sqla/migrate_repo/versions/49a21b4d4357_create_index_on_table_tokens.py +44 -0
  254. rucio/db/sqla/migrate_repo/versions/4a2cbedda8b9_add_source_replica_expression_column_to_.py +43 -0
  255. rucio/db/sqla/migrate_repo/versions/4a7182d9578b_added_bytes_length_accessed_at_columns.py +49 -0
  256. rucio/db/sqla/migrate_repo/versions/4bab9edd01fc_create_index_on_requests_rule_id.py +40 -0
  257. rucio/db/sqla/migrate_repo/versions/4c3a4acfe006_new_attr_account_table.py +63 -0
  258. rucio/db/sqla/migrate_repo/versions/4cf0a2e127d4_adding_transient_metadata.py +43 -0
  259. rucio/db/sqla/migrate_repo/versions/4df2c5ddabc0_remove_temporary_dids.py +55 -0
  260. rucio/db/sqla/migrate_repo/versions/50280c53117c_add_qos_class_to_rse.py +45 -0
  261. rucio/db/sqla/migrate_repo/versions/52153819589c_add_rse_id_to_replicas_table.py +43 -0
  262. rucio/db/sqla/migrate_repo/versions/52fd9f4916fa_added_activity_to_rules.py +43 -0
  263. rucio/db/sqla/migrate_repo/versions/53b479c3cb0f_fix_did_meta_table_missing_updated_at_.py +45 -0
  264. rucio/db/sqla/migrate_repo/versions/5673b4b6e843_add_wfms_metadata_to_rule_tables.py +47 -0
  265. rucio/db/sqla/migrate_repo/versions/575767d9f89_added_source_history_table.py +58 -0
  266. rucio/db/sqla/migrate_repo/versions/58bff7008037_add_started_at_to_requests.py +45 -0
  267. rucio/db/sqla/migrate_repo/versions/58c8b78301ab_rename_callback_to_message.py +106 -0
  268. rucio/db/sqla/migrate_repo/versions/5f139f77382a_added_child_rule_id_column.py +55 -0
  269. rucio/db/sqla/migrate_repo/versions/688ef1840840_adding_did_meta_table.py +50 -0
  270. rucio/db/sqla/migrate_repo/versions/6e572a9bfbf3_add_new_split_container_column_to_rules.py +47 -0
  271. rucio/db/sqla/migrate_repo/versions/70587619328_add_comment_column_for_subscriptions.py +43 -0
  272. rucio/db/sqla/migrate_repo/versions/739064d31565_remove_history_table_pks.py +41 -0
  273. rucio/db/sqla/migrate_repo/versions/7541902bf173_add_didsfollowed_and_followevents_table.py +91 -0
  274. rucio/db/sqla/migrate_repo/versions/7ec22226cdbf_new_replica_state_for_temporary_.py +72 -0
  275. rucio/db/sqla/migrate_repo/versions/810a41685bc1_added_columns_rse_transfer_limits.py +49 -0
  276. rucio/db/sqla/migrate_repo/versions/83f991c63a93_correct_rse_expression_length.py +43 -0
  277. rucio/db/sqla/migrate_repo/versions/8523998e2e76_increase_size_of_extended_attributes_.py +43 -0
  278. rucio/db/sqla/migrate_repo/versions/8ea9122275b1_adding_missing_function_based_indices.py +53 -0
  279. rucio/db/sqla/migrate_repo/versions/90f47792bb76_add_clob_payload_to_messages.py +45 -0
  280. rucio/db/sqla/migrate_repo/versions/914b8f02df38_new_table_for_lifetime_model_exceptions.py +68 -0
  281. rucio/db/sqla/migrate_repo/versions/94a5961ddbf2_add_estimator_columns.py +45 -0
  282. rucio/db/sqla/migrate_repo/versions/9a1b149a2044_add_saml_identity_type.py +94 -0
  283. rucio/db/sqla/migrate_repo/versions/9a45bc4ea66d_add_vp_table.py +54 -0
  284. rucio/db/sqla/migrate_repo/versions/9eb936a81eb1_true_is_true.py +72 -0
  285. rucio/db/sqla/migrate_repo/versions/a08fa8de1545_transfer_stats_table.py +55 -0
  286. rucio/db/sqla/migrate_repo/versions/a118956323f8_added_vo_table_and_vo_col_to_rse.py +76 -0
  287. rucio/db/sqla/migrate_repo/versions/a193a275255c_add_status_column_in_messages.py +47 -0
  288. rucio/db/sqla/migrate_repo/versions/a5f6f6e928a7_1_7_0.py +121 -0
  289. rucio/db/sqla/migrate_repo/versions/a616581ee47_added_columns_to_table_requests.py +59 -0
  290. rucio/db/sqla/migrate_repo/versions/a6eb23955c28_state_idx_non_functional.py +52 -0
  291. rucio/db/sqla/migrate_repo/versions/a74275a1ad30_added_global_quota_table.py +54 -0
  292. rucio/db/sqla/migrate_repo/versions/a93e4e47bda_heartbeats.py +64 -0
  293. rucio/db/sqla/migrate_repo/versions/ae2a56fcc89_added_comment_column_to_rules.py +49 -0
  294. rucio/db/sqla/migrate_repo/versions/b0070f3695c8_add_deletedidmeta_table.py +57 -0
  295. rucio/db/sqla/migrate_repo/versions/b4293a99f344_added_column_identity_to_table_tokens.py +43 -0
  296. rucio/db/sqla/migrate_repo/versions/b5493606bbf5_fix_primary_key_for_subscription_history.py +41 -0
  297. rucio/db/sqla/migrate_repo/versions/b7d287de34fd_removal_of_replicastate_source.py +91 -0
  298. rucio/db/sqla/migrate_repo/versions/b818052fa670_add_index_to_quarantined_replicas.py +40 -0
  299. rucio/db/sqla/migrate_repo/versions/b8caac94d7f0_add_comments_column_for_subscriptions_.py +43 -0
  300. rucio/db/sqla/migrate_repo/versions/b96a1c7e1cc4_new_bad_pfns_table_and_bad_replicas_.py +143 -0
  301. rucio/db/sqla/migrate_repo/versions/bb695f45c04_extend_request_state.py +76 -0
  302. rucio/db/sqla/migrate_repo/versions/bc68e9946deb_add_staging_timestamps_to_request.py +50 -0
  303. rucio/db/sqla/migrate_repo/versions/bf3baa1c1474_correct_pk_and_idx_for_history_tables.py +72 -0
  304. rucio/db/sqla/migrate_repo/versions/c0937668555f_add_qos_policy_map_table.py +55 -0
  305. rucio/db/sqla/migrate_repo/versions/c129ccdb2d5_add_lumiblocknr_to_dids.py +43 -0
  306. rucio/db/sqla/migrate_repo/versions/ccdbcd48206e_add_did_type_column_index_on_did_meta_.py +65 -0
  307. rucio/db/sqla/migrate_repo/versions/cebad904c4dd_new_payload_column_for_heartbeats.py +47 -0
  308. rucio/db/sqla/migrate_repo/versions/d1189a09c6e0_oauth2_0_and_jwt_feature_support_adding_.py +146 -0
  309. rucio/db/sqla/migrate_repo/versions/d23453595260_extend_request_state_for_preparer.py +104 -0
  310. rucio/db/sqla/migrate_repo/versions/d6dceb1de2d_added_purge_column_to_rules.py +44 -0
  311. rucio/db/sqla/migrate_repo/versions/d6e2c3b2cf26_remove_third_party_copy_column_from_rse.py +43 -0
  312. rucio/db/sqla/migrate_repo/versions/d91002c5841_new_account_limits_table.py +103 -0
  313. rucio/db/sqla/migrate_repo/versions/e138c364ebd0_extending_columns_for_filter_and_.py +49 -0
  314. rucio/db/sqla/migrate_repo/versions/e59300c8b179_support_for_archive.py +104 -0
  315. rucio/db/sqla/migrate_repo/versions/f1b14a8c2ac1_postgres_use_check_constraints.py +29 -0
  316. rucio/db/sqla/migrate_repo/versions/f41ffe206f37_oracle_global_temporary_tables.py +74 -0
  317. rucio/db/sqla/migrate_repo/versions/f85a2962b021_adding_transfertool_column_to_requests_.py +47 -0
  318. rucio/db/sqla/migrate_repo/versions/fa7a7d78b602_increase_refresh_token_size.py +43 -0
  319. rucio/db/sqla/migrate_repo/versions/fb28a95fe288_add_replicas_rse_id_tombstone_idx.py +37 -0
  320. rucio/db/sqla/migrate_repo/versions/fe1a65b176c9_set_third_party_copy_read_and_write_.py +43 -0
  321. rucio/db/sqla/migrate_repo/versions/fe8ea2fa9788_added_third_party_copy_column_to_rse_.py +43 -0
  322. rucio/db/sqla/models.py +1743 -0
  323. rucio/db/sqla/sautils.py +55 -0
  324. rucio/db/sqla/session.py +529 -0
  325. rucio/db/sqla/types.py +206 -0
  326. rucio/db/sqla/util.py +543 -0
  327. rucio/gateway/__init__.py +13 -0
  328. rucio/gateway/account.py +345 -0
  329. rucio/gateway/account_limit.py +363 -0
  330. rucio/gateway/authentication.py +381 -0
  331. rucio/gateway/config.py +227 -0
  332. rucio/gateway/credential.py +70 -0
  333. rucio/gateway/did.py +987 -0
  334. rucio/gateway/dirac.py +83 -0
  335. rucio/gateway/exporter.py +60 -0
  336. rucio/gateway/heartbeat.py +76 -0
  337. rucio/gateway/identity.py +189 -0
  338. rucio/gateway/importer.py +46 -0
  339. rucio/gateway/lifetime_exception.py +121 -0
  340. rucio/gateway/lock.py +153 -0
  341. rucio/gateway/meta_conventions.py +98 -0
  342. rucio/gateway/permission.py +74 -0
  343. rucio/gateway/quarantined_replica.py +79 -0
  344. rucio/gateway/replica.py +538 -0
  345. rucio/gateway/request.py +330 -0
  346. rucio/gateway/rse.py +632 -0
  347. rucio/gateway/rule.py +437 -0
  348. rucio/gateway/scope.py +100 -0
  349. rucio/gateway/subscription.py +280 -0
  350. rucio/gateway/vo.py +126 -0
  351. rucio/rse/__init__.py +96 -0
  352. rucio/rse/protocols/__init__.py +13 -0
  353. rucio/rse/protocols/bittorrent.py +194 -0
  354. rucio/rse/protocols/cache.py +111 -0
  355. rucio/rse/protocols/dummy.py +100 -0
  356. rucio/rse/protocols/gfal.py +708 -0
  357. rucio/rse/protocols/globus.py +243 -0
  358. rucio/rse/protocols/http_cache.py +82 -0
  359. rucio/rse/protocols/mock.py +123 -0
  360. rucio/rse/protocols/ngarc.py +209 -0
  361. rucio/rse/protocols/posix.py +250 -0
  362. rucio/rse/protocols/protocol.py +361 -0
  363. rucio/rse/protocols/rclone.py +365 -0
  364. rucio/rse/protocols/rfio.py +145 -0
  365. rucio/rse/protocols/srm.py +338 -0
  366. rucio/rse/protocols/ssh.py +414 -0
  367. rucio/rse/protocols/storm.py +195 -0
  368. rucio/rse/protocols/webdav.py +594 -0
  369. rucio/rse/protocols/xrootd.py +302 -0
  370. rucio/rse/rsemanager.py +881 -0
  371. rucio/rse/translation.py +260 -0
  372. rucio/tests/__init__.py +13 -0
  373. rucio/tests/common.py +280 -0
  374. rucio/tests/common_server.py +149 -0
  375. rucio/transfertool/__init__.py +13 -0
  376. rucio/transfertool/bittorrent.py +200 -0
  377. rucio/transfertool/bittorrent_driver.py +50 -0
  378. rucio/transfertool/bittorrent_driver_qbittorrent.py +134 -0
  379. rucio/transfertool/fts3.py +1600 -0
  380. rucio/transfertool/fts3_plugins.py +152 -0
  381. rucio/transfertool/globus.py +201 -0
  382. rucio/transfertool/globus_library.py +181 -0
  383. rucio/transfertool/mock.py +89 -0
  384. rucio/transfertool/transfertool.py +221 -0
  385. rucio/vcsversion.py +11 -0
  386. rucio/version.py +45 -0
  387. rucio/web/__init__.py +13 -0
  388. rucio/web/rest/__init__.py +13 -0
  389. rucio/web/rest/flaskapi/__init__.py +13 -0
  390. rucio/web/rest/flaskapi/authenticated_bp.py +27 -0
  391. rucio/web/rest/flaskapi/v1/__init__.py +13 -0
  392. rucio/web/rest/flaskapi/v1/accountlimits.py +236 -0
  393. rucio/web/rest/flaskapi/v1/accounts.py +1103 -0
  394. rucio/web/rest/flaskapi/v1/archives.py +102 -0
  395. rucio/web/rest/flaskapi/v1/auth.py +1644 -0
  396. rucio/web/rest/flaskapi/v1/common.py +426 -0
  397. rucio/web/rest/flaskapi/v1/config.py +304 -0
  398. rucio/web/rest/flaskapi/v1/credentials.py +213 -0
  399. rucio/web/rest/flaskapi/v1/dids.py +2340 -0
  400. rucio/web/rest/flaskapi/v1/dirac.py +116 -0
  401. rucio/web/rest/flaskapi/v1/export.py +75 -0
  402. rucio/web/rest/flaskapi/v1/heartbeats.py +127 -0
  403. rucio/web/rest/flaskapi/v1/identities.py +285 -0
  404. rucio/web/rest/flaskapi/v1/import.py +132 -0
  405. rucio/web/rest/flaskapi/v1/lifetime_exceptions.py +312 -0
  406. rucio/web/rest/flaskapi/v1/locks.py +358 -0
  407. rucio/web/rest/flaskapi/v1/main.py +91 -0
  408. rucio/web/rest/flaskapi/v1/meta_conventions.py +241 -0
  409. rucio/web/rest/flaskapi/v1/metrics.py +36 -0
  410. rucio/web/rest/flaskapi/v1/nongrid_traces.py +97 -0
  411. rucio/web/rest/flaskapi/v1/ping.py +88 -0
  412. rucio/web/rest/flaskapi/v1/redirect.py +366 -0
  413. rucio/web/rest/flaskapi/v1/replicas.py +1894 -0
  414. rucio/web/rest/flaskapi/v1/requests.py +998 -0
  415. rucio/web/rest/flaskapi/v1/rses.py +2250 -0
  416. rucio/web/rest/flaskapi/v1/rules.py +854 -0
  417. rucio/web/rest/flaskapi/v1/scopes.py +159 -0
  418. rucio/web/rest/flaskapi/v1/subscriptions.py +650 -0
  419. rucio/web/rest/flaskapi/v1/templates/auth_crash.html +80 -0
  420. rucio/web/rest/flaskapi/v1/templates/auth_granted.html +82 -0
  421. rucio/web/rest/flaskapi/v1/traces.py +137 -0
  422. rucio/web/rest/flaskapi/v1/types.py +20 -0
  423. rucio/web/rest/flaskapi/v1/vos.py +278 -0
  424. rucio/web/rest/main.py +18 -0
  425. rucio/web/rest/metrics.py +27 -0
  426. rucio/web/rest/ping.py +27 -0
  427. rucio-37.0.0rc1.data/data/rucio/etc/alembic.ini.template +71 -0
  428. rucio-37.0.0rc1.data/data/rucio/etc/alembic_offline.ini.template +74 -0
  429. rucio-37.0.0rc1.data/data/rucio/etc/globus-config.yml.template +5 -0
  430. rucio-37.0.0rc1.data/data/rucio/etc/ldap.cfg.template +30 -0
  431. rucio-37.0.0rc1.data/data/rucio/etc/mail_templates/rule_approval_request.tmpl +38 -0
  432. rucio-37.0.0rc1.data/data/rucio/etc/mail_templates/rule_approved_admin.tmpl +4 -0
  433. rucio-37.0.0rc1.data/data/rucio/etc/mail_templates/rule_approved_user.tmpl +17 -0
  434. rucio-37.0.0rc1.data/data/rucio/etc/mail_templates/rule_denied_admin.tmpl +6 -0
  435. rucio-37.0.0rc1.data/data/rucio/etc/mail_templates/rule_denied_user.tmpl +17 -0
  436. rucio-37.0.0rc1.data/data/rucio/etc/mail_templates/rule_ok_notification.tmpl +19 -0
  437. rucio-37.0.0rc1.data/data/rucio/etc/rse-accounts.cfg.template +25 -0
  438. rucio-37.0.0rc1.data/data/rucio/etc/rucio.cfg.atlas.client.template +43 -0
  439. rucio-37.0.0rc1.data/data/rucio/etc/rucio.cfg.template +241 -0
  440. rucio-37.0.0rc1.data/data/rucio/etc/rucio_multi_vo.cfg.template +217 -0
  441. rucio-37.0.0rc1.data/data/rucio/requirements.server.txt +297 -0
  442. rucio-37.0.0rc1.data/data/rucio/tools/bootstrap.py +34 -0
  443. rucio-37.0.0rc1.data/data/rucio/tools/merge_rucio_configs.py +144 -0
  444. rucio-37.0.0rc1.data/data/rucio/tools/reset_database.py +40 -0
  445. rucio-37.0.0rc1.data/scripts/rucio +133 -0
  446. rucio-37.0.0rc1.data/scripts/rucio-abacus-account +74 -0
  447. rucio-37.0.0rc1.data/scripts/rucio-abacus-collection-replica +46 -0
  448. rucio-37.0.0rc1.data/scripts/rucio-abacus-rse +78 -0
  449. rucio-37.0.0rc1.data/scripts/rucio-admin +97 -0
  450. rucio-37.0.0rc1.data/scripts/rucio-atropos +60 -0
  451. rucio-37.0.0rc1.data/scripts/rucio-auditor +206 -0
  452. rucio-37.0.0rc1.data/scripts/rucio-automatix +50 -0
  453. rucio-37.0.0rc1.data/scripts/rucio-bb8 +57 -0
  454. rucio-37.0.0rc1.data/scripts/rucio-cache-client +141 -0
  455. rucio-37.0.0rc1.data/scripts/rucio-cache-consumer +42 -0
  456. rucio-37.0.0rc1.data/scripts/rucio-conveyor-finisher +58 -0
  457. rucio-37.0.0rc1.data/scripts/rucio-conveyor-poller +66 -0
  458. rucio-37.0.0rc1.data/scripts/rucio-conveyor-preparer +37 -0
  459. rucio-37.0.0rc1.data/scripts/rucio-conveyor-receiver +44 -0
  460. rucio-37.0.0rc1.data/scripts/rucio-conveyor-stager +76 -0
  461. rucio-37.0.0rc1.data/scripts/rucio-conveyor-submitter +139 -0
  462. rucio-37.0.0rc1.data/scripts/rucio-conveyor-throttler +104 -0
  463. rucio-37.0.0rc1.data/scripts/rucio-dark-reaper +53 -0
  464. rucio-37.0.0rc1.data/scripts/rucio-dumper +160 -0
  465. rucio-37.0.0rc1.data/scripts/rucio-follower +44 -0
  466. rucio-37.0.0rc1.data/scripts/rucio-hermes +54 -0
  467. rucio-37.0.0rc1.data/scripts/rucio-judge-cleaner +89 -0
  468. rucio-37.0.0rc1.data/scripts/rucio-judge-evaluator +137 -0
  469. rucio-37.0.0rc1.data/scripts/rucio-judge-injector +44 -0
  470. rucio-37.0.0rc1.data/scripts/rucio-judge-repairer +44 -0
  471. rucio-37.0.0rc1.data/scripts/rucio-kronos +44 -0
  472. rucio-37.0.0rc1.data/scripts/rucio-minos +53 -0
  473. rucio-37.0.0rc1.data/scripts/rucio-minos-temporary-expiration +50 -0
  474. rucio-37.0.0rc1.data/scripts/rucio-necromancer +120 -0
  475. rucio-37.0.0rc1.data/scripts/rucio-oauth-manager +63 -0
  476. rucio-37.0.0rc1.data/scripts/rucio-reaper +83 -0
  477. rucio-37.0.0rc1.data/scripts/rucio-replica-recoverer +248 -0
  478. rucio-37.0.0rc1.data/scripts/rucio-rse-decommissioner +66 -0
  479. rucio-37.0.0rc1.data/scripts/rucio-storage-consistency-actions +74 -0
  480. rucio-37.0.0rc1.data/scripts/rucio-transmogrifier +77 -0
  481. rucio-37.0.0rc1.data/scripts/rucio-undertaker +76 -0
  482. rucio-37.0.0rc1.dist-info/METADATA +92 -0
  483. rucio-37.0.0rc1.dist-info/RECORD +487 -0
  484. rucio-37.0.0rc1.dist-info/WHEEL +5 -0
  485. rucio-37.0.0rc1.dist-info/licenses/AUTHORS.rst +100 -0
  486. rucio-37.0.0rc1.dist-info/licenses/LICENSE +201 -0
  487. rucio-37.0.0rc1.dist-info/top_level.txt +1 -0
@@ -0,0 +1,1144 @@
1
+ # Copyright European Organization for Nuclear Research (CERN) since 2012
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+
15
+ from typing import TYPE_CHECKING
16
+
17
+ import rucio.core.scope
18
+ from rucio.common.constants import RseAttr
19
+ from rucio.core.account import has_account_attribute, list_account_attributes
20
+ from rucio.core.identity import exist_identity_account
21
+ from rucio.core.lifetime_exception import list_exceptions
22
+ from rucio.core.rse import list_rse_attributes
23
+ from rucio.core.rse_expression_parser import parse_expression
24
+ from rucio.core.rule import get_rule
25
+ from rucio.db.sqla.constants import IdentityType
26
+
27
+ if TYPE_CHECKING:
28
+ from typing import Optional
29
+
30
+ from sqlalchemy.orm import Session
31
+
32
+ from rucio.common.types import InternalAccount
33
+
34
+
35
+ def has_permission(issuer, action, kwargs, *, session: "Optional[Session]" = None):
36
+ """
37
+ Checks if an account has the specified permission to
38
+ execute an action with parameters.
39
+
40
+ :param issuer: Account identifier which issues the command..
41
+ :param action: The action(API call) called by the account.
42
+ :param kwargs: List of arguments for the action.
43
+ :param session: The DB session to use
44
+ :returns: True if account is allowed, otherwise False
45
+ """
46
+ perm = {'add_account': perm_add_account,
47
+ 'del_account': perm_del_account,
48
+ 'update_account': perm_update_account,
49
+ 'add_rule': perm_add_rule,
50
+ 'add_subscription': perm_add_subscription,
51
+ 'add_scope': perm_add_scope,
52
+ 'add_rse': perm_add_rse,
53
+ 'update_rse': perm_update_rse,
54
+ 'add_protocol': perm_add_protocol,
55
+ 'del_protocol': perm_del_protocol,
56
+ 'update_protocol': perm_update_protocol,
57
+ 'declare_bad_file_replicas': perm_declare_bad_file_replicas,
58
+ 'declare_suspicious_file_replicas': perm_declare_suspicious_file_replicas,
59
+ 'add_replicas': perm_add_replicas,
60
+ 'delete_replicas': perm_delete_replicas,
61
+ 'skip_availability_check': perm_skip_availability_check,
62
+ 'update_replicas_states': perm_update_replicas_states,
63
+ 'add_rse_attribute': perm_add_rse_attribute,
64
+ 'del_rse_attribute': perm_del_rse_attribute,
65
+ 'del_rse': perm_del_rse,
66
+ 'del_rule': perm_del_rule,
67
+ 'update_rule': perm_update_rule,
68
+ 'approve_rule': perm_approve_rule,
69
+ 'update_subscription': perm_update_subscription,
70
+ 'reduce_rule': perm_reduce_rule,
71
+ 'move_rule': perm_move_rule,
72
+ 'get_auth_token_user_pass': perm_get_auth_token_user_pass,
73
+ 'get_auth_token_gss': perm_get_auth_token_gss,
74
+ 'get_auth_token_x509': perm_get_auth_token_x509,
75
+ 'get_auth_token_saml': perm_get_auth_token_saml,
76
+ 'add_account_identity': perm_add_account_identity,
77
+ 'add_did': perm_add_did,
78
+ 'add_dids': perm_add_dids,
79
+ 'attach_dids': perm_attach_dids,
80
+ 'detach_dids': perm_detach_dids,
81
+ 'attach_dids_to_dids': perm_attach_dids_to_dids,
82
+ 'create_did_sample': perm_create_did_sample,
83
+ 'set_metadata': perm_set_metadata,
84
+ 'set_status': perm_set_status,
85
+ 'queue_requests': perm_queue_requests,
86
+ 'set_rse_usage': perm_set_rse_usage,
87
+ 'set_rse_limits': perm_set_rse_limits,
88
+ 'list_requests': perm_list_requests,
89
+ 'list_requests_history': perm_list_requests_history,
90
+ 'get_request_by_did': perm_get_request_by_did,
91
+ 'get_request_history_by_did': perm_get_request_history_by_did,
92
+ 'cancel_request': perm_cancel_request,
93
+ 'get_next': perm_get_next,
94
+ 'set_local_account_limit': perm_set_local_account_limit,
95
+ 'set_global_account_limit': perm_set_global_account_limit,
96
+ 'delete_local_account_limit': perm_delete_local_account_limit,
97
+ 'delete_global_account_limit': perm_delete_global_account_limit,
98
+ 'config_sections': perm_config,
99
+ 'config_add_section': perm_config,
100
+ 'config_has_section': perm_config,
101
+ 'config_options': perm_config,
102
+ 'config_has_option': perm_config,
103
+ 'config_get': perm_config,
104
+ 'config_items': perm_config,
105
+ 'config_set': perm_config,
106
+ 'config_remove_section': perm_config,
107
+ 'config_remove_option': perm_config,
108
+ 'get_local_account_usage': perm_get_local_account_usage,
109
+ 'get_global_account_usage': perm_get_global_account_usage,
110
+ 'add_attribute': perm_add_account_attribute,
111
+ 'del_attribute': perm_del_account_attribute,
112
+ 'list_heartbeats': perm_list_heartbeats,
113
+ 'resurrect': perm_resurrect,
114
+ 'update_lifetime_exceptions': perm_update_lifetime_exceptions,
115
+ 'get_auth_token_ssh': perm_get_auth_token_ssh,
116
+ 'get_signed_url': perm_get_signed_url,
117
+ 'add_bad_pfns': perm_add_bad_pfns,
118
+ 'del_account_identity': perm_del_account_identity,
119
+ 'del_identity': perm_del_identity,
120
+ 'remove_did_from_followed': perm_remove_did_from_followed,
121
+ 'remove_dids_from_followed': perm_remove_dids_from_followed,
122
+ 'add_vo': perm_add_vo,
123
+ 'list_vos': perm_list_vos,
124
+ 'recover_vo_root_identity': perm_recover_vo_root_identity,
125
+ 'update_vo': perm_update_vo,
126
+ 'access_rule_vo': perm_access_rule_vo}
127
+
128
+ return perm.get(action, perm_default)(issuer=issuer, kwargs=kwargs, session=session)
129
+
130
+
131
+ def _is_root(issuer):
132
+ return issuer.external == 'root'
133
+
134
+
135
+ def perm_default(issuer, kwargs, *, session: "Optional[Session]" = None):
136
+ """
137
+ Default permission.
138
+
139
+ :param issuer: Account identifier which issues the command.
140
+ :param kwargs: List of arguments for the action.
141
+ :param session: The DB session to use
142
+ :returns: True if account is allowed, otherwise False
143
+ """
144
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
145
+
146
+
147
+ def perm_add_rse(issuer, kwargs, *, session: "Optional[Session]" = None):
148
+ """
149
+ Checks if an account can add a RSE.
150
+
151
+ :param issuer: Account identifier which issues the command.
152
+ :param kwargs: List of arguments for the action.
153
+ :param session: The DB session to use
154
+ :returns: True if account is allowed, otherwise False
155
+ """
156
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
157
+
158
+
159
+ def perm_update_rse(issuer, kwargs, *, session: "Optional[Session]" = None):
160
+ """
161
+ Checks if an account can update a RSE.
162
+
163
+ :param issuer: Account identifier which issues the command.
164
+ :param kwargs: List of arguments for the action.
165
+ :param session: The DB session to use
166
+ :returns: True if account is allowed, otherwise False
167
+ """
168
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
169
+
170
+
171
+ def perm_add_rule(issuer, kwargs, *, session: "Optional[Session]" = None):
172
+ """
173
+ Checks if an account can add a replication rule.
174
+
175
+ :param issuer: Account identifier which issues the command.
176
+ :param kwargs: List of arguments for the action.
177
+ :param session: The DB session to use
178
+ :returns: True if account is allowed, otherwise False
179
+ """
180
+ if kwargs['account'] == issuer and not kwargs['locked']:
181
+ return True
182
+ if _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session):
183
+ return True
184
+ return False
185
+
186
+
187
+ def perm_add_subscription(issuer, kwargs, *, session: "Optional[Session]" = None):
188
+ """
189
+ Checks if an account can add a subscription.
190
+
191
+ :param issuer: Account identifier which issues the command.
192
+ :param kwargs: List of arguments for the action.
193
+ :param session: The DB session to use
194
+ :returns: True if account is allowed, otherwise False
195
+ """
196
+ if _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session):
197
+ return True
198
+ return False
199
+
200
+
201
+ def perm_add_rse_attribute(issuer, kwargs, *, session: "Optional[Session]" = None):
202
+ """
203
+ Checks if an account can add a RSE attribute.
204
+
205
+ :param issuer: Account identifier which issues the command.
206
+ :param kwargs: List of arguments for the action.
207
+ :param session: The DB session to use
208
+ :returns: True if account is allowed, otherwise False
209
+ """
210
+ if _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session):
211
+ return True
212
+ return False
213
+
214
+
215
+ def perm_del_rse_attribute(issuer, kwargs, *, session: "Optional[Session]" = None):
216
+ """
217
+ Checks if an account can delete a RSE attribute.
218
+
219
+ :param issuer: Account identifier which issues the command.
220
+ :param kwargs: List of arguments for the action.
221
+ :param session: The DB session to use
222
+ :returns: True if account is allowed, otherwise False
223
+ """
224
+ if _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session):
225
+ return True
226
+ return False
227
+
228
+
229
+ def perm_del_rse(issuer, kwargs, *, session: "Optional[Session]" = None):
230
+ """
231
+ Checks if an account can delete a RSE.
232
+
233
+ :param issuer: Account identifier which issues the command.
234
+ :param kwargs: List of arguments for the action.
235
+ :param session: The DB session to use
236
+ :returns: True if account is allowed, otherwise False
237
+ """
238
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
239
+
240
+
241
+ def perm_add_account(issuer, kwargs, *, session: "Optional[Session]" = None):
242
+ """
243
+ Checks if an account can add an account.
244
+
245
+ :param issuer: Account identifier which issues the command.
246
+ :param kwargs: List of arguments for the action.
247
+ :param session: The DB session to use
248
+ :returns: True if account is allowed, otherwise False
249
+ """
250
+ return _is_root(issuer)
251
+
252
+
253
+ def perm_del_account(issuer, kwargs, *, session: "Optional[Session]" = None):
254
+ """
255
+ Checks if an account can del an account.
256
+
257
+ :param issuer: Account identifier which issues the command.
258
+ :param kwargs: List of arguments for the action.
259
+ :param session: The DB session to use
260
+ :returns: True if account is allowed, otherwise False
261
+ """
262
+ return _is_root(issuer)
263
+
264
+
265
+ def perm_update_account(issuer, kwargs, *, session: "Optional[Session]" = None):
266
+ """
267
+ Checks if an account can update an account.
268
+
269
+ :param issuer: Account identifier which issues the command.
270
+ :param kwargs: List of arguments for the action.
271
+ :param session: The DB session to use
272
+ :returns: True if account is allowed, otherwise False
273
+ """
274
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
275
+
276
+
277
+ def perm_add_scope(issuer, kwargs, *, session: "Optional[Session]" = None):
278
+ """
279
+ Checks if an account can add a scop to a account.
280
+
281
+ :param issuer: Account identifier which issues the command.
282
+ :param kwargs: List of arguments for the action.
283
+ :param session: The DB session to use
284
+ :returns: True if account is allowed, otherwise False
285
+ """
286
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
287
+
288
+
289
+ def perm_get_auth_token_user_pass(issuer, kwargs, *, session: "Optional[Session]" = None):
290
+ """
291
+ Checks if a user can request a token with user_pass for an account.
292
+
293
+ :param issuer: Account identifier which issues the command.
294
+ :param kwargs: List of arguments for the action.
295
+ :param session: The DB session to use
296
+ :returns: True if account is allowed, otherwise False
297
+ """
298
+ if exist_identity_account(identity=kwargs['username'], type_=IdentityType.USERPASS, account=kwargs['account'], session=session):
299
+ return True
300
+ return False
301
+
302
+
303
+ def perm_get_auth_token_gss(issuer, kwargs, *, session: "Optional[Session]" = None):
304
+ """
305
+ Checks if a user can request a token with user_pass for an account.
306
+
307
+ :param issuer: Account identifier which issues the command.
308
+ :param kwargs: List of arguments for the action.
309
+ :param session: The DB session to use
310
+ :returns: True if account is allowed, otherwise False
311
+ """
312
+ if exist_identity_account(identity=kwargs['gsscred'], type_=IdentityType.GSS, account=kwargs['account'], session=session):
313
+ return True
314
+ return False
315
+
316
+
317
+ def perm_get_auth_token_x509(issuer, kwargs, *, session: "Optional[Session]" = None):
318
+ """
319
+ Checks if a user can request a token with user_pass for an account.
320
+
321
+ :param issuer: Account identifier which issues the command.
322
+ :param kwargs: List of arguments for the action.
323
+ :param session: The DB session to use
324
+ :returns: True if account is allowed, otherwise False
325
+ """
326
+ if exist_identity_account(identity=kwargs['dn'], type_=IdentityType.X509, account=kwargs['account'], session=session):
327
+ return True
328
+ return False
329
+
330
+
331
+ def perm_get_auth_token_saml(issuer, kwargs, *, session: "Optional[Session]" = None):
332
+ """
333
+ Checks if a user can request a token with user_pass for an account.
334
+
335
+ :param issuer: Account identifier which issues the command.
336
+ :param kwargs: List of arguments for the action.
337
+ :param session: The DB session to use
338
+ :returns: True if account is allowed, otherwise False
339
+ """
340
+ if exist_identity_account(identity=kwargs['saml_nameid'], type_=IdentityType.SAML, account=kwargs['account'], session=session):
341
+ return True
342
+ return False
343
+
344
+
345
+ def perm_add_account_identity(issuer, kwargs, *, session: "Optional[Session]" = None):
346
+ """
347
+ Checks if an account can add an identity to an account.
348
+
349
+ :param issuer: Account identifier which issues the command.
350
+ :param kwargs: List of arguments for the action.
351
+ :param session: The DB session to use
352
+ :returns: True if account is allowed, otherwise False
353
+ """
354
+
355
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
356
+
357
+
358
+ def perm_del_account_identity(issuer, kwargs, *, session: "Optional[Session]" = None):
359
+ """
360
+ Checks if an account can delete an identity to an account.
361
+
362
+ :param issuer: Account identifier which issues the command.
363
+ :param kwargs: List of arguments for the action.
364
+ :param session: The DB session to use
365
+ :returns: True if account is allowed, otherwise False
366
+ """
367
+
368
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
369
+
370
+
371
+ def perm_del_identity(issuer, kwargs, *, session: "Optional[Session]" = None):
372
+ """
373
+ Checks if an account can delete an identity.
374
+
375
+ :param issuer: Account identifier which issues the command.
376
+ :param kwargs: List of arguments for the action.
377
+ :param session: The DB session to use
378
+ :returns: True if account is allowed, otherwise False
379
+ """
380
+
381
+ return _is_root(issuer) or issuer.external in kwargs.get('accounts')
382
+
383
+
384
+ def perm_add_did(issuer, kwargs, *, session: "Optional[Session]" = None):
385
+ """
386
+ Checks if an account can add an data identifier to a scope.
387
+
388
+ :param issuer: Account identifier which issues the command.
389
+ :param kwargs: List of arguments for the action.
390
+ :param session: The DB session to use
391
+ :returns: True if account is allowed, otherwise False
392
+ """
393
+ # Check the accounts of the issued rules
394
+ if not _is_root(issuer) and not has_account_attribute(account=issuer, key='admin', session=session):
395
+ for rule in kwargs.get('rules', []):
396
+ if rule['account'] != issuer:
397
+ return False
398
+
399
+ return _is_root(issuer)\
400
+ or has_account_attribute(account=issuer, key='admin', session=session)\
401
+ or rucio.core.scope.is_scope_owner(scope=kwargs['scope'], account=issuer, session=session)\
402
+ or kwargs['scope'].external == 'mock'
403
+
404
+
405
+ def perm_add_dids(issuer, kwargs, *, session: "Optional[Session]" = None):
406
+ """
407
+ Checks if an account can bulk add data identifiers.
408
+
409
+ :param issuer: Account identifier which issues the command.
410
+ :param kwargs: List of arguments for the action.
411
+ :param session: The DB session to use
412
+ :returns: True if account is allowed, otherwise False
413
+ """
414
+ # Check the accounts of the issued rules
415
+ return all(perm_add_did(issuer, kwargs=did, session=session) for did in kwargs['dids'])
416
+
417
+
418
+ def perm_attach_dids(issuer, kwargs, *, session: "Optional[Session]" = None):
419
+ """
420
+ Checks if an account can append an data identifier to the other data identifier.
421
+
422
+ :param issuer: Account identifier which issues the command.
423
+ :param kwargs: List of arguments for the action.
424
+ :param session: The DB session to use
425
+ :returns: True if account is allowed, otherwise False
426
+ """
427
+ return _is_root(issuer)\
428
+ or has_account_attribute(account=issuer, key='admin', session=session)\
429
+ or rucio.core.scope.is_scope_owner(scope=kwargs['scope'], account=issuer, session=session)\
430
+ or kwargs['scope'].external == 'mock'
431
+
432
+
433
+ def perm_attach_dids_to_dids(issuer, kwargs, *, session: "Optional[Session]" = None):
434
+ """
435
+ Checks if an account can append an data identifier to the other data identifier.
436
+
437
+ :param issuer: Account identifier which issues the command.
438
+ :param kwargs: List of arguments for the action.
439
+ :param session: The DB session to use
440
+ :returns: True if account is allowed, otherwise False
441
+ """
442
+ if _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session):
443
+ return True
444
+ else:
445
+ attachments = kwargs['attachments']
446
+ scopes = [did['scope'] for did in attachments]
447
+ scopes = list(set(scopes))
448
+ for scope in scopes:
449
+ if not rucio.core.scope.is_scope_owner(scope, issuer, session=session):
450
+ return False
451
+ return True
452
+
453
+
454
+ def perm_create_did_sample(issuer, kwargs, *, session: "Optional[Session]" = None):
455
+ """
456
+ Checks if an account can create a sample of a data identifier collection.
457
+
458
+ :param issuer: Account identifier which issues the command.
459
+ :param kwargs: List of arguments for the action.
460
+ :param session: The DB session to use
461
+ :returns: True if account is allowed, otherwise False
462
+ """
463
+ return _is_root(issuer)\
464
+ or has_account_attribute(account=issuer, key='admin', session=session)\
465
+ or rucio.core.scope.is_scope_owner(scope=kwargs['scope'], account=issuer, session=session)\
466
+ or kwargs['scope'].external == 'mock'
467
+
468
+
469
+ def perm_del_rule(issuer, kwargs, *, session: "Optional[Session]" = None):
470
+ """
471
+ Checks if an issuer can delete a replication rule.
472
+
473
+ :param issuer: Account identifier which issues the command.
474
+ :param kwargs: List of arguments for the action.
475
+ :param session: The DB session to use
476
+ :returns: True if account is allowed to call the API call, otherwise False
477
+ """
478
+ if _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session):
479
+ return True
480
+ return False
481
+
482
+
483
+ def perm_update_rule(issuer, kwargs, *, session: "Optional[Session]" = None):
484
+ """
485
+ Checks if an issuer can update a replication rule.
486
+
487
+ :param issuer: Account identifier which issues the command.
488
+ :param kwargs: List of arguments for the action.
489
+ :param session: The DB session to use
490
+ :returns: True if account is allowed to call the API call, otherwise False
491
+ """
492
+ if _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session):
493
+ return True
494
+ return False
495
+
496
+
497
+ def perm_approve_rule(issuer, kwargs, *, session: "Optional[Session]" = None):
498
+ """
499
+ Checks if an issuer can approve a replication rule.
500
+
501
+ :param issuer: Account identifier which issues the command.
502
+ :param kwargs: List of arguments for the action.
503
+ :param session: The DB session to use
504
+ :returns: True if account is allowed to call the API call, otherwise False
505
+ """
506
+ if _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session):
507
+ return True
508
+ return False
509
+
510
+
511
+ def perm_reduce_rule(issuer, kwargs, *, session: "Optional[Session]" = None):
512
+ """
513
+ Checks if an issuer can reduce a replication rule.
514
+
515
+ :param issuer: Account identifier which issues the command.
516
+ :param kwargs: List of arguments for the action.
517
+ :param session: The DB session to use
518
+ :returns: True if account is allowed to call the API call, otherwise False
519
+ """
520
+ if _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session):
521
+ return True
522
+ return False
523
+
524
+
525
+ def perm_move_rule(issuer, kwargs, *, session: "Optional[Session]" = None):
526
+ """
527
+ Checks if an issuer can move a replication rule.
528
+
529
+ :param issuer: Account identifier which issues the command.
530
+ :param kwargs: List of arguments for the action.
531
+ :param session: The DB session to use
532
+ :returns: True if account is allowed to call the API call, otherwise False
533
+ """
534
+ if _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session):
535
+ return True
536
+ return False
537
+
538
+
539
+ def perm_update_subscription(issuer, kwargs, *, session: "Optional[Session]" = None):
540
+ """
541
+ Checks if an account can update a subscription.
542
+
543
+ :param issuer: Account identifier which issues the command.
544
+ :param kwargs: List of arguments for the action.
545
+ :param session: The DB session to use
546
+ :returns: True if account is allowed, otherwise False
547
+ """
548
+ if _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session):
549
+ return True
550
+
551
+ return False
552
+
553
+
554
+ def perm_detach_dids(issuer, kwargs, *, session: "Optional[Session]" = None):
555
+ """
556
+ Checks if an account can detach an data identifier from the other data identifier.
557
+
558
+ :param issuer: Account identifier which issues the command.
559
+ :param kwargs: List of arguments for the action.
560
+ :param session: The DB session to use
561
+ :returns: True if account is allowed, otherwise False
562
+ """
563
+ return perm_attach_dids(issuer, kwargs, session=session)
564
+
565
+
566
+ def perm_set_metadata(issuer, kwargs, *, session: "Optional[Session]" = None):
567
+ """
568
+ Checks if an account can set a metadata on a data identifier.
569
+
570
+ :param issuer: Account identifier which issues the command.
571
+ :param kwargs: List of arguments for the action.
572
+ :param session: The DB session to use
573
+ :returns: True if account is allowed, otherwise False
574
+ """
575
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session) or rucio.core.scope.is_scope_owner(scope=kwargs['scope'], account=issuer, session=session)
576
+
577
+
578
+ def perm_set_status(issuer, kwargs, *, session: "Optional[Session]" = None):
579
+ """
580
+ Checks if an account can set status on an data identifier.
581
+
582
+ :param issuer: Account identifier which issues the command.
583
+ :param kwargs: List of arguments for the action.
584
+ :param session: The DB session to use
585
+ :returns: True if account is allowed, otherwise False
586
+ """
587
+ if kwargs.get('open', False):
588
+ if not _is_root(issuer) and not has_account_attribute(account=issuer, key='admin', session=session):
589
+ return False
590
+
591
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session) or rucio.core.scope.is_scope_owner(scope=kwargs['scope'], account=issuer, session=session)
592
+
593
+
594
+ def perm_add_protocol(issuer, kwargs, *, session: "Optional[Session]" = None):
595
+ """
596
+ Checks if an account can add a protocol to an RSE.
597
+
598
+ :param issuer: Account identifier which issues the command.
599
+ :param kwargs: List of arguments for the action.
600
+ :param session: The DB session to use
601
+ :returns: True if account is allowed, otherwise False
602
+ """
603
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
604
+
605
+
606
+ def perm_del_protocol(issuer, kwargs, *, session: "Optional[Session]" = None):
607
+ """
608
+ Checks if an account can delete protocols from an RSE.
609
+
610
+ :param issuer: Account identifier which issues the command.
611
+ :param kwargs: List of arguments for the action.
612
+ :param session: The DB session to use
613
+ :returns: True if account is allowed, otherwise False
614
+ """
615
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
616
+
617
+
618
+ def perm_update_protocol(issuer, kwargs, *, session: "Optional[Session]" = None):
619
+ """
620
+ Checks if an account can update protocols of an RSE.
621
+
622
+ :param issuer: Account identifier which issues the command.
623
+ :param kwargs: List of arguments for the action.
624
+ :param session: The DB session to use
625
+ :returns: True if account is allowed, otherwise False
626
+ """
627
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
628
+
629
+
630
+ def perm_declare_bad_file_replicas(issuer, kwargs, *, session: "Optional[Session]" = None):
631
+ """
632
+ Checks if an account can declare bad file replicas.
633
+
634
+ :param issuer: Account identifier which issues the command.
635
+ :param kwargs: List of arguments for the action.
636
+ :param session: The DB session to use
637
+ :returns: True if account is allowed, otherwise False
638
+ """
639
+ return _is_root(issuer)
640
+
641
+
642
+ def perm_declare_suspicious_file_replicas(issuer, kwargs, *, session: "Optional[Session]" = None):
643
+ """
644
+ Checks if an account can declare suspicious file replicas.
645
+
646
+ :param issuer: Account identifier which issues the command.
647
+ :param kwargs: List of arguments for the action.
648
+ :param session: The DB session to use
649
+ :returns: True if account is allowed, otherwise False
650
+ """
651
+ return True
652
+
653
+
654
+ def perm_add_replicas(issuer, kwargs, *, session: "Optional[Session]" = None):
655
+ """
656
+ Checks if an account can add replicas.
657
+
658
+ :param issuer: Account identifier which issues the command.
659
+ :param kwargs: List of arguments for the action.
660
+ :param session: The DB session to use
661
+ :returns: True if account is allowed, otherwise False
662
+ """
663
+ return str(kwargs.get('rse', '')).endswith('SCRATCHDISK')\
664
+ or str(kwargs.get('rse', '')).endswith('USERDISK')\
665
+ or str(kwargs.get('rse', '')).endswith('MOCK')\
666
+ or str(kwargs.get('rse', '')).endswith('LOCALGROUPDISK')\
667
+ or _is_root(issuer)\
668
+ or has_account_attribute(account=issuer, key='admin', session=session)
669
+
670
+
671
+ def perm_skip_availability_check(issuer, kwargs, *, session: "Optional[Session]" = None):
672
+ """
673
+ Checks if an account can skip the availabity check to add/delete file replicas.
674
+
675
+ :param issuer: Account identifier which issues the command.
676
+ :param kwargs: List of arguments for the action.
677
+ :param session: The DB session to use
678
+ :returns: True if account is allowed, otherwise False
679
+ """
680
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
681
+
682
+
683
+ def perm_delete_replicas(issuer, kwargs, *, session: "Optional[Session]" = None):
684
+ """
685
+ Checks if an account can delete replicas.
686
+
687
+ :param issuer: Account identifier which issues the command.
688
+ :param kwargs: List of arguments for the action.
689
+ :param session: The DB session to use
690
+ :returns: True if account is allowed, otherwise False
691
+ """
692
+ return False
693
+
694
+
695
+ def perm_update_replicas_states(issuer, kwargs, *, session: "Optional[Session]" = None):
696
+ """
697
+ Checks if an account can delete replicas.
698
+
699
+ :param issuer: Account identifier which issues the command.
700
+ :param kwargs: List of arguments for the action.
701
+ :param session: The DB session to use
702
+ :returns: True if account is allowed, otherwise False
703
+ """
704
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
705
+
706
+
707
+ def perm_queue_requests(issuer, kwargs, *, session: "Optional[Session]" = None):
708
+ """
709
+ Checks if an account can submit transfer or deletion requests on destination RSEs for data identifiers.
710
+
711
+ :param issuer: Account identifier which issues the command.
712
+ :param kwargs: List of arguments for the action.
713
+ :param session: The DB session to use
714
+ :returns: True if account is allowed, otherwise False
715
+ """
716
+ return _is_root(issuer)
717
+
718
+
719
+ def perm_list_requests(issuer, kwargs, *, session: "Optional[Session]" = None):
720
+ """
721
+ Checks if an account can list requests.
722
+
723
+ :param issuer: Account identifier which issues the command.
724
+ :param kwargs: List of arguments for the action.
725
+ :param session: The DB session to use
726
+ :returns: True if account is allowed, otherwise False
727
+ """
728
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
729
+
730
+
731
+ def perm_list_requests_history(issuer, kwargs, *, session: "Optional[Session]" = None):
732
+ """
733
+ Checks if an account can list historical requests.
734
+
735
+ :param issuer: Account identifier which issues the command.
736
+ :param kwargs: List of arguments for the action.
737
+ :param session: The DB session to use
738
+ :returns: True if account is allowed, otherwise False
739
+ """
740
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
741
+
742
+
743
+ def perm_get_request_by_did(issuer, kwargs, *, session: "Optional[Session]" = None):
744
+ """
745
+ Checks if an account can get a request by DID.
746
+
747
+ :param issuer: Account identifier which issues the command.
748
+ :param kwargs: List of arguments for the action.
749
+ :param session: The DB session to use
750
+ :returns: True if account is allowed, otherwise False
751
+ """
752
+ return True
753
+
754
+
755
+ def perm_get_request_history_by_did(issuer, kwargs, *, session: "Optional[Session]" = None):
756
+ """
757
+ Checks if an account can get a historical request by DID.
758
+
759
+ :param issuer: Account identifier which issues the command.
760
+ :param kwargs: List of arguments for the action.
761
+ :param session: The DB session to use
762
+ :returns: True if account is allowed, otherwise False
763
+ """
764
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
765
+
766
+
767
+ def perm_cancel_request(issuer, kwargs, *, session: "Optional[Session]" = None):
768
+ """
769
+ Checks if an account can cancel a request.
770
+
771
+ :param issuer: Account identifier which issues the command.
772
+ :param kwargs: List of arguments for the action.
773
+ :param session: The DB session to use
774
+ :returns: True if account is allowed, otherwise False
775
+ """
776
+ return _is_root(issuer)
777
+
778
+
779
+ def perm_get_next(issuer, kwargs, *, session: "Optional[Session]" = None):
780
+ """
781
+ Checks if an account can retrieve the next request matching the request type and state.
782
+
783
+ :param issuer: Account identifier which issues the command.
784
+ :param kwargs: List of arguments for the action.
785
+ :param session: The DB session to use
786
+ :returns: True if account is allowed, otherwise False
787
+ """
788
+ return _is_root(issuer)
789
+
790
+
791
+ def perm_set_rse_usage(issuer, kwargs, *, session: "Optional[Session]" = None):
792
+ """
793
+ Checks if an account can set RSE usage information.
794
+
795
+ :param issuer: Account identifier which issues the command.
796
+ :param kwargs: List of arguments for the action.
797
+ :param session: The DB session to use
798
+ :returns: True if account is allowed to call the API call, otherwise False
799
+ """
800
+ return _is_root(issuer)
801
+
802
+
803
+ def perm_set_rse_limits(issuer, kwargs, *, session: "Optional[Session]" = None):
804
+ """
805
+ Checks if an account can set RSE limits.
806
+
807
+ :param issuer: Account identifier which issues the command.
808
+ :param kwargs: List of arguments for the action.
809
+ :param session: The DB session to use
810
+ :returns: True if account is allowed to call the API call, otherwise False
811
+ """
812
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
813
+
814
+
815
+ def perm_set_local_account_limit(issuer, kwargs, *, session: "Optional[Session]" = None):
816
+ """
817
+ Checks if an account can set an account limit.
818
+
819
+ :param account: Account identifier which issues the command.
820
+ :param kwargs: List of arguments for the action.
821
+ :param session: The DB session to use
822
+ :returns: True if account is allowed, otherwise False
823
+ """
824
+ if _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session):
825
+ return True
826
+ # Check if user is a country admin
827
+ admin_in_country = []
828
+ for kv in list_account_attributes(account=issuer, session=session):
829
+ if kv['key'].startswith('country-') and kv['value'] == 'admin':
830
+ admin_in_country.append(kv['key'].partition('-')[2])
831
+ if admin_in_country and list_rse_attributes(rse_id=kwargs['rse_id'], session=session).get(RseAttr.COUNTRY) in admin_in_country:
832
+ return True
833
+ return False
834
+
835
+
836
+ def perm_set_global_account_limit(issuer, kwargs, *, session: "Optional[Session]" = None):
837
+ """
838
+ Checks if an account can set a global account limit.
839
+
840
+ :param account: Account identifier which issues the command.
841
+ :param kwargs: List of arguments for the action.
842
+ :param session: The DB session to use
843
+ :returns: True if account is allowed, otherwise False
844
+ """
845
+ if _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session):
846
+ return True
847
+ # Check if user is a country admin
848
+ admin_in_country = set()
849
+ for kv in list_account_attributes(account=issuer, session=session):
850
+ if kv['key'].startswith('country-') and kv['value'] == 'admin':
851
+ admin_in_country.add(kv['key'].partition('-')[2])
852
+ resolved_rse_countries = {list_rse_attributes(rse_id=rse['rse_id'], session=session).get(RseAttr.COUNTRY)
853
+ for rse in parse_expression(kwargs['rse_expression'], filter_={'vo': issuer.vo}, session=session)}
854
+ if resolved_rse_countries.issubset(admin_in_country):
855
+ return True
856
+ return False
857
+
858
+
859
+ def perm_delete_local_account_limit(issuer, kwargs, *, session: "Optional[Session]" = None):
860
+ """
861
+ Checks if an account can delete an account limit.
862
+
863
+ :param account: Account identifier which issues the command.
864
+ :param kwargs: List of arguments for the action.
865
+ :param session: The DB session to use
866
+ :returns: True if account is allowed, otherwise False
867
+ """
868
+ if _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session):
869
+ return True
870
+ # Check if user is a country admin
871
+ admin_in_country = []
872
+ for kv in list_account_attributes(account=issuer, session=session):
873
+ if kv['key'].startswith('country-') and kv['value'] == 'admin':
874
+ admin_in_country.append(kv['key'].partition('-')[2])
875
+ if admin_in_country and list_rse_attributes(rse_id=kwargs['rse_id'], session=session).get(RseAttr.COUNTRY) in admin_in_country:
876
+ return True
877
+ return False
878
+
879
+
880
+ def perm_delete_global_account_limit(issuer, kwargs, *, session: "Optional[Session]" = None):
881
+ """
882
+ Checks if an account can delete a global account limit.
883
+
884
+ :param issuer: Account identifier which issues the command.
885
+ :param kwargs: List of arguments for the action.
886
+ :param session: The DB session to use
887
+ :returns: True if account is allowed, otherwise False
888
+ """
889
+ if _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session):
890
+ return True
891
+ # Check if user is a country admin
892
+ admin_in_country = set()
893
+ for kv in list_account_attributes(account=issuer, session=session):
894
+ if kv['key'].startswith('country-') and kv['value'] == 'admin':
895
+ admin_in_country.add(kv['key'].partition('-')[2])
896
+ if admin_in_country:
897
+ resolved_rse_countries = {list_rse_attributes(rse_id=rse['rse_id'], session=session).get(RseAttr.COUNTRY)
898
+ for rse in parse_expression(kwargs['rse_expression'], filter_={'vo': issuer.vo}, session=session)}
899
+ if resolved_rse_countries.issubset(admin_in_country):
900
+ return True
901
+ return False
902
+
903
+
904
+ def perm_config(issuer, kwargs, *, session: "Optional[Session]" = None):
905
+ """
906
+ Checks if an account can read/write the configuration.
907
+
908
+ :param issuer: Account identifier which issues the command.
909
+ :param kwargs: List of arguments for the action.
910
+ :param session: The DB session to use
911
+ :returns: True if account is allowed to call the API call, otherwise False
912
+ """
913
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
914
+
915
+
916
+ def perm_get_local_account_usage(issuer, kwargs, *, session: "Optional[Session]" = None):
917
+ """
918
+ Checks if an account can get the account usage of an account.
919
+
920
+ :param issuer: Account identifier which issues the command.
921
+ :param kwargs: List of arguments for the action.
922
+ :param session: The DB session to use
923
+ :returns: True if account is allowed, otherwise False
924
+ """
925
+ if _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session) or kwargs.get('account') == issuer:
926
+ return True
927
+ # Check if user is a country admin
928
+ for kv in list_account_attributes(account=issuer, session=session):
929
+ if kv['key'].startswith('country-') and kv['value'] == 'admin':
930
+ return True
931
+ return False
932
+
933
+
934
+ def perm_get_global_account_usage(issuer, kwargs, *, session: "Optional[Session]" = None):
935
+ """
936
+ Checks if an account can get the account usage of an account.
937
+
938
+ :param issuer: Account identifier which issues the command.
939
+ :param kwargs: List of arguments for the action.
940
+ :param session: The DB session to use
941
+ :returns: True if account is allowed, otherwise False
942
+ """
943
+ if _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session) or kwargs.get('account') == issuer:
944
+ return True
945
+
946
+ # Check if user is a country admin for all involved countries
947
+ admin_in_country = set()
948
+ for kv in list_account_attributes(account=issuer, session=session):
949
+ if kv['key'].startswith('country-') and kv['value'] == 'admin':
950
+ admin_in_country.add(kv['key'].partition('-')[2])
951
+ resolved_rse_countries = {list_rse_attributes(rse_id=rse['rse_id'], session=session).get(RseAttr.COUNTRY)
952
+ for rse in parse_expression(kwargs['rse_exp'], filter_={'vo': issuer.vo}, session=session)}
953
+
954
+ if resolved_rse_countries.issubset(admin_in_country):
955
+ return True
956
+ return False
957
+
958
+
959
+ def perm_add_account_attribute(issuer, kwargs, *, session: "Optional[Session]" = None):
960
+ """
961
+ Checks if an account can add attributes to accounts.
962
+
963
+ :param issuer: Account identifier which issues the command.
964
+ :param kwargs: List of arguments for the action.
965
+ :param session: The DB session to use
966
+ :returns: True if account is allowed to call the API call, otherwise False
967
+ """
968
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
969
+
970
+
971
+ def perm_del_account_attribute(issuer, kwargs, *, session: "Optional[Session]" = None):
972
+ """
973
+ Checks if an account can add attributes to accounts.
974
+
975
+ :param issuer: Account identifier which issues the command.
976
+ :param kwargs: List of arguments for the action.
977
+ :param session: The DB session to use
978
+ :returns: True if account is allowed to call the API call, otherwise False
979
+ """
980
+ return perm_add_account_attribute(issuer, kwargs, session=session)
981
+
982
+
983
+ def perm_list_heartbeats(issuer, kwargs, *, session: "Optional[Session]" = None):
984
+ """
985
+ Checks if an account can list heartbeats.
986
+
987
+ :param issuer: Account identifier which issues the command.
988
+ :param kwargs: List of arguments for the action.
989
+ :param session: The DB session to use
990
+ :returns: True if account is allowed to call the API call, otherwise False
991
+ """
992
+ return _is_root(issuer)
993
+
994
+
995
+ def perm_resurrect(issuer, kwargs, *, session: "Optional[Session]" = None):
996
+ """
997
+ Checks if an account can resurrect DIDS.
998
+
999
+ :param issuer: Account identifier which issues the command.
1000
+ :param kwargs: List of arguments for the action.
1001
+ :param session: The DB session to use
1002
+ :returns: True if account is allowed to call the API call, otherwise False
1003
+ """
1004
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
1005
+
1006
+
1007
+ def perm_update_lifetime_exceptions(issuer, kwargs, *, session: "Optional[Session]" = None):
1008
+ """
1009
+ Checks if an account can approve/reject Lifetime Model exceptions.
1010
+
1011
+ :param issuer: Account identifier which issues the command.
1012
+ :param session: The DB session to use
1013
+ :returns: True if account is allowed to call the API call, otherwise False
1014
+ """
1015
+ if kwargs['vo'] is not None:
1016
+ exceptions = next(list_exceptions(exception_id=kwargs['exception_id'], states=False, session=session))
1017
+ if exceptions['scope'].vo != kwargs['vo']:
1018
+ return False
1019
+ return _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session)
1020
+
1021
+
1022
+ def perm_get_auth_token_ssh(issuer: "InternalAccount", kwargs: dict, *, session: "Optional[Session]" = None) -> bool:
1023
+ """
1024
+ Checks if an account can request an ssh token.
1025
+
1026
+ :param issuer: Account identifier which issues the command.
1027
+ :param session: The DB session to use
1028
+ :returns: True if account is allowed to call the API call, otherwise False
1029
+ """
1030
+ return True
1031
+
1032
+
1033
+ def perm_get_signed_url(issuer, kwargs, *, session: "Optional[Session]" = None):
1034
+ """
1035
+ Checks if an account can request a signed URL.
1036
+
1037
+ :param issuer: Account identifier which issues the command.
1038
+ :param session: The DB session to use
1039
+ :returns: True if account is allowed to call the API call, otherwise False
1040
+ """
1041
+ return _is_root(issuer)
1042
+
1043
+
1044
+ def perm_add_bad_pfns(issuer, kwargs, *, session: "Optional[Session]" = None):
1045
+ """
1046
+ Checks if an account can declare bad PFNs.
1047
+
1048
+ :param issuer: Account identifier which issues the command.
1049
+ :param kwargs: List of arguments for the action.
1050
+ :param session: The DB session to use
1051
+ :returns: True if account is allowed, otherwise False
1052
+ """
1053
+ return _is_root(issuer)
1054
+
1055
+
1056
+ def perm_remove_did_from_followed(issuer, kwargs, *, session: "Optional[Session]" = None):
1057
+ """
1058
+ Checks if an account can remove did from followed table.
1059
+
1060
+ :param issuer: Account identifier which issues the command.
1061
+ :param kwargs: List of arguments for the action.
1062
+ :param session: The DB session to use
1063
+ :returns: True if account is allowed, otherwise False
1064
+ """
1065
+ return _is_root(issuer)\
1066
+ or has_account_attribute(account=issuer, key='admin', session=session)\
1067
+ or kwargs['account'] == issuer\
1068
+ or kwargs['scope'].external == 'mock'
1069
+
1070
+
1071
+ def perm_remove_dids_from_followed(issuer, kwargs, *, session: "Optional[Session]" = None):
1072
+ """
1073
+ Checks if an account can bulk remove dids from followed table.
1074
+
1075
+ :param issuer: Account identifier which issues the command.
1076
+ :param kwargs: List of arguments for the action.
1077
+ :param session: The DB session to use
1078
+ :returns: True if account is allowed, otherwise False
1079
+ """
1080
+ if _is_root(issuer) or has_account_attribute(account=issuer, key='admin', session=session):
1081
+ return True
1082
+ if not kwargs['account'] == issuer:
1083
+ return False
1084
+ return True
1085
+
1086
+
1087
+ def perm_add_vo(issuer, kwargs, *, session: "Optional[Session]" = None):
1088
+ """
1089
+ Checks if an account can add a VO.
1090
+
1091
+ :param issuer: Account identifier which issues the command.
1092
+ :param kwargs: List of arguments for the action.
1093
+ :param session: The DB session to use
1094
+ :returns: True if account is allowed, otherwise False
1095
+ """
1096
+ return (issuer.internal == 'super_root')
1097
+
1098
+
1099
+ def perm_list_vos(issuer, kwargs, *, session: "Optional[Session]" = None):
1100
+ """
1101
+ Checks if an account can list a VO.
1102
+
1103
+ :param issuer: Account identifier which issues the command.
1104
+ :param kwargs: List of arguments for the action.
1105
+ :param session: The DB session to use
1106
+ :returns: True if account is allowed, otherwise False
1107
+ """
1108
+ return (issuer.internal == 'super_root')
1109
+
1110
+
1111
+ def perm_recover_vo_root_identity(issuer, kwargs, *, session: "Optional[Session]" = None):
1112
+ """
1113
+ Checks if an account can recover identities for VOs.
1114
+
1115
+ :param issuer: Account identifier which issues the command.
1116
+ :param kwargs: List of arguments for the action.
1117
+ :param session: The DB session to use
1118
+ :returns: True if account is allowed, otherwise False
1119
+ """
1120
+ return (issuer.internal == 'super_root')
1121
+
1122
+
1123
+ def perm_update_vo(issuer, kwargs, *, session: "Optional[Session]" = None):
1124
+ """
1125
+ Checks if an account can update a VO.
1126
+
1127
+ :param issuer: Account identifier which issues the command.
1128
+ :param kwargs: List of arguments for the action.
1129
+ :param session: The DB session to use
1130
+ :returns: True if account is allowed, otherwise False
1131
+ """
1132
+ return (issuer.internal == 'super_root')
1133
+
1134
+
1135
+ def perm_access_rule_vo(issuer, kwargs, *, session: "Optional[Session]" = None):
1136
+ """
1137
+ Checks if we're at the same VO as the rule_id's
1138
+
1139
+ :param issuer: Account identifier which issues the command.
1140
+ :param kwargs: List of arguments for the action.
1141
+ :param session: The DB session to use
1142
+ :returns: True if account is allowed, otherwise False
1143
+ """
1144
+ return get_rule(kwargs['rule_id'])['scope'].vo == issuer.vo