mage-ai 0.9.45__py3-none-any.whl → 0.9.47__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.
Files changed (566) hide show
  1. mage_ai/api/constants.py +2 -0
  2. mage_ai/api/operations/base.py +230 -12
  3. mage_ai/api/policies/BasePolicy.py +3 -3
  4. mage_ai/api/policies/ComputeClusterPolicy.py +96 -0
  5. mage_ai/api/policies/ComputeConnectionPolicy.py +74 -0
  6. mage_ai/api/policies/ComputeServicePolicy.py +50 -0
  7. mage_ai/api/policies/DownloadPolicy.py +29 -0
  8. mage_ai/api/policies/GlobalHookPolicy.py +126 -0
  9. mage_ai/api/policies/IntegrationSourcePolicy.py +62 -0
  10. mage_ai/api/policies/OauthPolicy.py +56 -4
  11. mage_ai/api/policies/ProjectPolicy.py +1 -0
  12. mage_ai/api/presenters/ComputeClusterPresenter.py +19 -0
  13. mage_ai/api/presenters/ComputeConnectionPresenter.py +26 -0
  14. mage_ai/api/presenters/ComputeServicePresenter.py +15 -0
  15. mage_ai/api/presenters/DownloadPresenter.py +13 -0
  16. mage_ai/api/presenters/GlobalHookPresenter.py +74 -0
  17. mage_ai/api/presenters/IntegrationSourcePresenter.py +9 -2
  18. mage_ai/api/presenters/PipelinePresenter.py +3 -0
  19. mage_ai/api/presenters/PipelineRunPresenter.py +8 -3
  20. mage_ai/api/presenters/PipelineSchedulePresenter.py +22 -1
  21. mage_ai/api/presenters/ProjectPresenter.py +1 -0
  22. mage_ai/api/presenters/SparkApplicationPresenter.py +2 -0
  23. mage_ai/api/presenters/WorkspacePresenter.py +22 -22
  24. mage_ai/api/resources/AsyncBaseResource.py +39 -0
  25. mage_ai/api/resources/BlockResource.py +5 -0
  26. mage_ai/api/resources/ClusterResource.py +1 -1
  27. mage_ai/api/resources/ComputeClusterResource.py +109 -0
  28. mage_ai/api/resources/ComputeConnectionResource.py +103 -0
  29. mage_ai/api/resources/ComputeServiceResource.py +35 -0
  30. mage_ai/api/resources/DataProviderResource.py +2 -0
  31. mage_ai/api/resources/DownloadResource.py +56 -0
  32. mage_ai/api/resources/ExecutionStateResource.py +1 -1
  33. mage_ai/api/resources/GlobalHookResource.py +192 -0
  34. mage_ai/api/resources/IntegrationSourceResource.py +149 -2
  35. mage_ai/api/resources/KernelResource.py +10 -0
  36. mage_ai/api/resources/OauthResource.py +28 -33
  37. mage_ai/api/resources/PipelineResource.py +4 -4
  38. mage_ai/api/resources/PipelineScheduleResource.py +37 -16
  39. mage_ai/api/resources/ProjectResource.py +5 -3
  40. mage_ai/api/resources/SessionResource.py +0 -9
  41. mage_ai/api/resources/SparkApplicationResource.py +5 -5
  42. mage_ai/api/resources/SparkEnvironmentResource.py +1 -2
  43. mage_ai/api/resources/SparkExecutorResource.py +1 -2
  44. mage_ai/api/resources/SparkJobResource.py +3 -6
  45. mage_ai/api/resources/SparkSqlResource.py +6 -11
  46. mage_ai/api/resources/SparkStageAttemptResource.py +2 -3
  47. mage_ai/api/resources/SparkStageAttemptTaskResource.py +1 -2
  48. mage_ai/api/resources/SparkStageAttemptTaskSummaryResource.py +1 -2
  49. mage_ai/api/resources/SparkStageResource.py +3 -6
  50. mage_ai/api/resources/SparkThreadResource.py +1 -2
  51. mage_ai/api/resources/mixins/spark.py +25 -4
  52. mage_ai/authentication/oauth2.py +1 -3
  53. mage_ai/authentication/permissions/constants.py +4 -0
  54. mage_ai/authentication/providers/active_directory.py +136 -0
  55. mage_ai/authentication/providers/constants.py +3 -0
  56. mage_ai/authentication/providers/ghe.py +24 -25
  57. mage_ai/authentication/providers/google.py +25 -26
  58. mage_ai/authentication/providers/oauth.py +2 -2
  59. mage_ai/authentication/providers/okta.py +28 -29
  60. mage_ai/authentication/providers/sso.py +2 -2
  61. mage_ai/cluster_manager/aws/emr_cluster_manager.py +2 -1
  62. mage_ai/cluster_manager/workspace/base.py +7 -3
  63. mage_ai/data_integrations/destinations/constants.py +1 -0
  64. mage_ai/data_integrations/sources/constants.py +5 -0
  65. mage_ai/data_integrations/utils/scheduler.py +57 -2
  66. mage_ai/data_preparation/executors/block_executor.py +9 -1
  67. mage_ai/data_preparation/executors/pipeline_executor.py +9 -0
  68. mage_ai/data_preparation/models/block/__init__.py +82 -17
  69. mage_ai/data_preparation/models/block/data_integration/constants.py +3 -0
  70. mage_ai/data_preparation/models/block/data_integration/data.py +1 -1
  71. mage_ai/data_preparation/models/block/data_integration/utils.py +196 -37
  72. mage_ai/data_preparation/models/block/spark/mixins.py +82 -34
  73. mage_ai/data_preparation/models/block/sql/utils/shared.py +18 -1
  74. mage_ai/data_preparation/models/block/utils.py +28 -13
  75. mage_ai/data_preparation/models/download/__init__.py +8 -0
  76. mage_ai/data_preparation/models/global_hooks/__init__.py +0 -0
  77. mage_ai/data_preparation/models/global_hooks/constants.py +93 -0
  78. mage_ai/data_preparation/models/global_hooks/models.py +992 -0
  79. mage_ai/data_preparation/models/global_hooks/predicates.py +316 -0
  80. mage_ai/data_preparation/models/global_hooks/utils.py +21 -0
  81. mage_ai/data_preparation/models/pipeline.py +84 -6
  82. mage_ai/data_preparation/models/pipelines/integration_pipeline.py +0 -1
  83. mage_ai/data_preparation/models/pipelines/models.py +16 -0
  84. mage_ai/data_preparation/models/project/__init__.py +6 -0
  85. mage_ai/data_preparation/models/project/constants.py +1 -0
  86. mage_ai/data_preparation/models/project/models.py +12 -0
  87. mage_ai/data_preparation/preferences.py +29 -18
  88. mage_ai/data_preparation/repo_manager.py +35 -3
  89. mage_ai/data_preparation/sync/__init__.py +2 -0
  90. mage_ai/data_preparation/templates/constants.py +14 -0
  91. mage_ai/data_preparation/templates/data_exporters/chroma.py +24 -0
  92. mage_ai/data_preparation/templates/data_exporters/streaming/rabbitmq.yaml +7 -0
  93. mage_ai/data_preparation/templates/data_loaders/chroma.py +27 -0
  94. mage_ai/data_preparation/templates/data_loaders/streaming/activemq.yaml +6 -0
  95. mage_ai/data_preparation/templates/data_loaders/streaming/nats.yaml +20 -0
  96. mage_ai/data_preparation/templates/repo/io_config.yaml +3 -0
  97. mage_ai/io/base.py +1 -0
  98. mage_ai/io/chroma.py +153 -0
  99. mage_ai/io/config.py +8 -0
  100. mage_ai/orchestration/db/models/schedules.py +100 -36
  101. mage_ai/orchestration/pipeline_scheduler.py +58 -19
  102. mage_ai/orchestration/queue/process_queue.py +9 -1
  103. mage_ai/orchestration/triggers/api.py +11 -3
  104. mage_ai/orchestration/triggers/constants.py +1 -0
  105. mage_ai/server/active_kernel.py +37 -4
  106. mage_ai/server/api/downloads.py +76 -1
  107. mage_ai/server/constants.py +1 -1
  108. mage_ai/server/frontend_dist/404.html +2 -2
  109. mage_ai/server/frontend_dist/_next/static/N3FS4bHv0jpYeeg672uYK/_buildManifest.js +1 -0
  110. mage_ai/server/frontend_dist/_next/static/chunks/1749-bf512b4dabbab7fa.js +1 -0
  111. mage_ai/server/frontend_dist/_next/static/chunks/{7519-8c29bbb92e03cc77.js → 1845-5ce774d5ab81ed57.js} +1 -1
  112. mage_ai/server/frontend_dist/_next/static/chunks/1952-57858e7445d24413.js +1 -0
  113. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/2714-1e79e9f2e998b544.js → frontend_dist/_next/static/chunks/2714-68fef54789d7eaeb.js} +1 -1
  114. mage_ai/server/frontend_dist/_next/static/chunks/2717-92cdffd87b6f6e05.js +1 -0
  115. mage_ai/server/frontend_dist/_next/static/chunks/3419-715ca383fa15a5ef.js +1 -0
  116. mage_ai/server/frontend_dist/_next/static/chunks/3437-b4d6a037cf5781f8.js +1 -0
  117. mage_ai/server/frontend_dist/_next/static/chunks/3943-c9fb980f03df6450.js +1 -0
  118. mage_ai/server/frontend_dist/_next/static/chunks/4267-cb102e060a43d9bd.js +1 -0
  119. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/3932-0ceca9599d6e6d00.js → frontend_dist/_next/static/chunks/4366-3e52497942acbafe.js} +1 -1
  120. mage_ai/server/frontend_dist/_next/static/chunks/{4783-1a21d9be47574bba.js → 4783-422429203610c318.js} +1 -1
  121. mage_ai/server/frontend_dist/_next/static/chunks/5457-949640f4037bf12f.js +1 -0
  122. mage_ai/server/frontend_dist/_next/static/chunks/5499-76cf8f023c6b0985.js +1 -0
  123. mage_ai/server/frontend_dist/_next/static/chunks/553-7f7919e14392ca67.js +1 -0
  124. mage_ai/server/frontend_dist/_next/static/chunks/5638-a65610405a70961c.js +1 -0
  125. mage_ai/server/frontend_dist/_next/static/chunks/5810-e26a0768db1cfdba.js +1 -0
  126. mage_ai/server/frontend_dist/_next/static/chunks/5820-28adeabb5cda2b96.js +1 -0
  127. mage_ai/server/frontend_dist/_next/static/chunks/{5896-14e5a23b1c6a0769.js → 5896-7b8e36634d7d94eb.js} +1 -1
  128. mage_ai/server/frontend_dist/_next/static/chunks/600-705fe234320ec5de.js +1 -0
  129. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/3122-f27de34d0b5426af.js → frontend_dist/_next/static/chunks/6285-648f9a732e100b2f.js} +1 -1
  130. mage_ai/server/frontend_dist/_next/static/chunks/6798-b904395b0c18647b.js +1 -0
  131. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/5397-b5f2e477acc6bd6b.js → frontend_dist/_next/static/chunks/6965-c613d1834c8ed92d.js} +1 -1
  132. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/7022-18fde36eb46e1d65.js → frontend_dist/_next/static/chunks/7022-e76cae3ba5ee5312.js} +1 -1
  133. mage_ai/server/frontend_dist/_next/static/chunks/722-900f786d24f91b2e.js +1 -0
  134. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/7361-694e1e4fb9c97d68.js → frontend_dist/_next/static/chunks/7361-6c5c9063b9f91700.js} +1 -1
  135. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/7858-c83d25349d0e980f.js → frontend_dist/_next/static/chunks/7858-d9df72e95e438284.js} +1 -1
  136. mage_ai/server/frontend_dist/_next/static/chunks/{8146-92e7ccfed169ee9c.js → 8146-27f0e31f309897a5.js} +1 -1
  137. mage_ai/server/frontend_dist/_next/static/chunks/8264-0d582a6ca33c3dfa.js +1 -0
  138. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/3684-cbda680fd8927d07.js → frontend_dist/_next/static/chunks/8432-f191e39f9b5893f2.js} +1 -1
  139. mage_ai/server/frontend_dist/_next/static/chunks/845-9a73c65fe3fdc328.js +1 -0
  140. mage_ai/server/frontend_dist/_next/static/chunks/8487-8e1c09546dff4dbf.js +1 -0
  141. mage_ai/server/frontend_dist/_next/static/chunks/90-a7308bae028d7001.js +1 -0
  142. mage_ai/server/frontend_dist/_next/static/chunks/9264-cc44b07f248707b0.js +1 -0
  143. mage_ai/server/frontend_dist/_next/static/chunks/9618-4eb49cdbd1ba11d7.js +1 -0
  144. mage_ai/server/frontend_dist/_next/static/chunks/{976-18c98af60b76f1a7.js → 976-0a8c2c4d7acd957b.js} +1 -1
  145. mage_ai/server/frontend_dist/_next/static/chunks/framework-22b71764bf44ede4.js +1 -0
  146. mage_ai/server/frontend_dist/_next/static/chunks/pages/_app-78c4a077a2f279c2.js +1 -0
  147. mage_ai/server/frontend_dist/_next/static/chunks/pages/block-layout-a24cb24b6f08bbc9.js +1 -0
  148. mage_ai/server/frontend_dist/_next/static/chunks/pages/compute-419775ca1293b354.js +1 -0
  149. mage_ai/server/frontend_dist/_next/static/chunks/pages/files-b37d221eb5ddc248.js +1 -0
  150. mage_ai/server/frontend_dist/_next/static/chunks/pages/global-data-products/{[...slug]-3f6fc312f67ff72e.js → [...slug]-cfd68e760ae00958.js} +1 -1
  151. mage_ai/server/frontend_dist/_next/static/chunks/pages/{global-data-products-f4cd03036c3e8723.js → global-data-products-c3b79ef31007f95b.js} +1 -1
  152. mage_ai/server/frontend_dist/_next/static/chunks/pages/global-hooks/[...slug]-7adc543fc490367a.js +1 -0
  153. mage_ai/server/frontend_dist/_next/static/chunks/pages/global-hooks-51d366993f6dd449.js +1 -0
  154. mage_ai/server/frontend_dist/_next/static/chunks/pages/manage/files-fe6e73463a20d67c.js +1 -0
  155. mage_ai/server/frontend_dist/_next/static/chunks/pages/manage/settings-60845f0b59142f32.js +1 -0
  156. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/manage/users/[user]-1d9b298fdceabbf1.js → frontend_dist/_next/static/chunks/pages/manage/users/[user]-9384c5f1efa2ac18.js} +1 -1
  157. mage_ai/server/frontend_dist/_next/static/chunks/pages/manage/users/new-abd8571907664fdf.js +1 -0
  158. mage_ai/server/frontend_dist/_next/static/chunks/pages/manage/{users-a7c970122a10afdc.js → users-28a930b148d99766.js} +1 -1
  159. mage_ai/server/frontend_dist/_next/static/chunks/pages/manage-d8a38b5d1f50e798.js +1 -0
  160. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/overview-a4647a274cf4dffa.js → frontend_dist/_next/static/chunks/pages/overview-f0c40645f385f23f.js} +1 -1
  161. mage_ai/server/frontend_dist/_next/static/chunks/pages/{pipeline-runs-c79819d6826e5416.js → pipeline-runs-b35d37bfba8fbccc.js} +1 -1
  162. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/backfills/[...slug]-9adc2974aada27ba.js +1 -0
  163. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/backfills-51b1311dff2a974e.js +1 -0
  164. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/dashboard-571c0962333b92f0.js +1 -0
  165. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/edit-2333e524d34b474a.js +1 -0
  166. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/{logs-a52d2d3e0c2978f4.js → logs-ef680455ae54ccbe.js} +1 -1
  167. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/monitors/block-runs-ddddcddd2f74b4f6.js +1 -0
  168. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/monitors/block-runtime-4a238307feddb522.js +1 -0
  169. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/monitors-6d1afeb4a84f50f7.js +1 -0
  170. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/runs/[run]-1ff9bb8e22ca1e75.js +1 -0
  171. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/runs-ce717786f31e8f04.js → frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/runs-2d20b2cd08907afd.js} +1 -1
  172. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/settings-11363aa58d51f4e1.js +1 -0
  173. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/syncs-b75bf17498e87354.js +1 -0
  174. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/triggers/[...slug]-779c3ef0676a12ac.js +1 -0
  175. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/triggers-bbea9a088657404a.js +1 -0
  176. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/{[pipeline]-35fe7762cb83a733.js → [pipeline]-02c843b9c8418bb5.js} +1 -1
  177. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines-3737f2b0afc2ede3.js +1 -0
  178. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/account/profile-3c8f062913c66f3e.js +1 -0
  179. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/permissions/{[...slug]-d6a62284c7c99cb3.js → [...slug]-b78b1be5b9ed84b9.js} +1 -1
  180. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/{permissions-b1389695f758c32b.js → permissions-37b78a436eeab258.js} +1 -1
  181. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/preferences-c1f4ed17d501ccca.js +1 -0
  182. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/roles/{[...slug]-0bdb66265286ab22.js → [...slug]-db05a80d18c168e5.js} +1 -1
  183. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/{roles-e04e18fc295ca76a.js → roles-f55c77e4f46c8d33.js} +1 -1
  184. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/sync-data-2a1f8737561fdd94.js +1 -0
  185. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/users/{[...slug]-0267358c91122109.js → [...slug]-e3bf6e5d8bb250c4.js} +1 -1
  186. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/{users-50fed18bb9b8d142.js → users-20f0a050a42a015d.js} +1 -1
  187. mage_ai/server/frontend_dist/_next/static/chunks/pages/{settings-56f83205752b1323.js → settings-0f0121db7f5ff93d.js} +1 -1
  188. mage_ai/server/frontend_dist/_next/static/chunks/pages/sign-in-99e2748e3c1d57a3.js +1 -0
  189. mage_ai/server/{frontend_dist_base_path_template/_next/static/chunks/pages/templates/[...slug]-9370551ffa462144.js → frontend_dist/_next/static/chunks/pages/templates/[...slug]-f44ccd1499ffd23a.js} +1 -1
  190. mage_ai/server/frontend_dist/_next/static/chunks/pages/{templates-3cff90df13a1a755.js → templates-1bfaa1c50e844813.js} +1 -1
  191. mage_ai/server/frontend_dist/_next/static/chunks/pages/terminal-ed121e305169cf1c.js +1 -0
  192. mage_ai/server/frontend_dist/_next/static/chunks/pages/test-9ae68758102cc843.js +1 -0
  193. mage_ai/server/frontend_dist/_next/static/chunks/pages/triggers-8bdd858240d5dbf6.js +1 -0
  194. mage_ai/server/frontend_dist/_next/static/chunks/pages/version-control-dd4fb405695f74bf.js +1 -0
  195. mage_ai/server/frontend_dist/_next/static/chunks/{webpack-8af2b89c5a3779fc.js → webpack-fea697dd168c6d0c.js} +1 -1
  196. mage_ai/server/frontend_dist/block-layout.html +2 -2
  197. mage_ai/server/frontend_dist/compute.html +5 -5
  198. mage_ai/server/frontend_dist/files.html +5 -5
  199. mage_ai/server/frontend_dist/global-data-products/[...slug].html +5 -5
  200. mage_ai/server/frontend_dist/global-data-products.html +5 -5
  201. mage_ai/server/frontend_dist/global-hooks/[...slug].html +24 -0
  202. mage_ai/server/frontend_dist/global-hooks.html +24 -0
  203. mage_ai/server/frontend_dist/index.html +2 -2
  204. mage_ai/server/frontend_dist/manage/files.html +5 -5
  205. mage_ai/server/frontend_dist/manage/settings.html +5 -5
  206. mage_ai/server/frontend_dist/manage/users/[user].html +5 -5
  207. mage_ai/server/frontend_dist/manage/users/new.html +5 -5
  208. mage_ai/server/frontend_dist/manage/users.html +5 -5
  209. mage_ai/server/frontend_dist/manage.html +5 -5
  210. mage_ai/server/frontend_dist/oauth.html +4 -4
  211. mage_ai/server/frontend_dist/overview.html +5 -5
  212. mage_ai/server/frontend_dist/pipeline-runs.html +5 -5
  213. mage_ai/server/frontend_dist/pipelines/[pipeline]/backfills/[...slug].html +5 -5
  214. mage_ai/server/frontend_dist/pipelines/[pipeline]/backfills.html +5 -5
  215. mage_ai/server/frontend_dist/pipelines/[pipeline]/dashboard.html +5 -5
  216. mage_ai/server/frontend_dist/pipelines/[pipeline]/edit.html +2 -2
  217. mage_ai/server/frontend_dist/pipelines/[pipeline]/logs.html +5 -5
  218. mage_ai/server/frontend_dist/pipelines/[pipeline]/monitors/block-runs.html +5 -5
  219. mage_ai/server/frontend_dist/pipelines/[pipeline]/monitors/block-runtime.html +5 -5
  220. mage_ai/server/frontend_dist/pipelines/[pipeline]/monitors.html +5 -5
  221. mage_ai/server/frontend_dist/pipelines/[pipeline]/runs/[run].html +5 -5
  222. mage_ai/server/frontend_dist/pipelines/[pipeline]/runs.html +5 -5
  223. mage_ai/server/frontend_dist/pipelines/[pipeline]/settings.html +5 -5
  224. mage_ai/server/frontend_dist/pipelines/[pipeline]/syncs.html +5 -5
  225. mage_ai/server/frontend_dist/pipelines/[pipeline]/triggers/[...slug].html +5 -5
  226. mage_ai/server/frontend_dist/pipelines/[pipeline]/triggers.html +5 -5
  227. mage_ai/server/frontend_dist/pipelines/[pipeline].html +2 -2
  228. mage_ai/server/frontend_dist/pipelines.html +5 -5
  229. mage_ai/server/frontend_dist/settings/account/profile.html +5 -5
  230. mage_ai/server/frontend_dist/settings/workspace/permissions/[...slug].html +5 -5
  231. mage_ai/server/frontend_dist/settings/workspace/permissions.html +5 -5
  232. mage_ai/server/frontend_dist/settings/workspace/preferences.html +5 -5
  233. mage_ai/server/frontend_dist/settings/workspace/roles/[...slug].html +5 -5
  234. mage_ai/server/frontend_dist/settings/workspace/roles.html +5 -5
  235. mage_ai/server/frontend_dist/settings/workspace/sync-data.html +5 -5
  236. mage_ai/server/frontend_dist/settings/workspace/users/[...slug].html +5 -5
  237. mage_ai/server/frontend_dist/settings/workspace/users.html +5 -5
  238. mage_ai/server/frontend_dist/settings.html +2 -2
  239. mage_ai/server/frontend_dist/sign-in.html +24 -25
  240. mage_ai/server/frontend_dist/templates/[...slug].html +5 -5
  241. mage_ai/server/frontend_dist/templates.html +5 -5
  242. mage_ai/server/frontend_dist/terminal.html +5 -5
  243. mage_ai/server/frontend_dist/test.html +15 -15
  244. mage_ai/server/frontend_dist/triggers.html +5 -5
  245. mage_ai/server/frontend_dist/version-control.html +5 -5
  246. mage_ai/server/frontend_dist_base_path_template/404.html +2 -2
  247. mage_ai/server/frontend_dist_base_path_template/_next/static/aoO6jYZLVlUGCCdY-wmy8/_buildManifest.js +1 -0
  248. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/1749-bf512b4dabbab7fa.js +1 -0
  249. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/{7519-8c29bbb92e03cc77.js → 1845-5ce774d5ab81ed57.js} +1 -1
  250. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/1952-57858e7445d24413.js +1 -0
  251. mage_ai/server/{frontend_dist/_next/static/chunks/2714-1e79e9f2e998b544.js → frontend_dist_base_path_template/_next/static/chunks/2714-68fef54789d7eaeb.js} +1 -1
  252. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/2717-92cdffd87b6f6e05.js +1 -0
  253. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/3419-715ca383fa15a5ef.js +1 -0
  254. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/3437-b4d6a037cf5781f8.js +1 -0
  255. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/3943-c9fb980f03df6450.js +1 -0
  256. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/4267-cb102e060a43d9bd.js +1 -0
  257. mage_ai/server/{frontend_dist/_next/static/chunks/3932-0ceca9599d6e6d00.js → frontend_dist_base_path_template/_next/static/chunks/4366-3e52497942acbafe.js} +1 -1
  258. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/{4783-1a21d9be47574bba.js → 4783-422429203610c318.js} +1 -1
  259. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/5457-949640f4037bf12f.js +1 -0
  260. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/5499-76cf8f023c6b0985.js +1 -0
  261. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/553-7f7919e14392ca67.js +1 -0
  262. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/5638-a65610405a70961c.js +1 -0
  263. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/5810-e26a0768db1cfdba.js +1 -0
  264. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/5820-28adeabb5cda2b96.js +1 -0
  265. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/{5896-14e5a23b1c6a0769.js → 5896-7b8e36634d7d94eb.js} +1 -1
  266. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/600-705fe234320ec5de.js +1 -0
  267. mage_ai/server/{frontend_dist/_next/static/chunks/3122-f27de34d0b5426af.js → frontend_dist_base_path_template/_next/static/chunks/6285-648f9a732e100b2f.js} +1 -1
  268. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/6798-b904395b0c18647b.js +1 -0
  269. mage_ai/server/{frontend_dist/_next/static/chunks/5397-b5f2e477acc6bd6b.js → frontend_dist_base_path_template/_next/static/chunks/6965-c613d1834c8ed92d.js} +1 -1
  270. mage_ai/server/{frontend_dist/_next/static/chunks/7022-18fde36eb46e1d65.js → frontend_dist_base_path_template/_next/static/chunks/7022-e76cae3ba5ee5312.js} +1 -1
  271. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/722-900f786d24f91b2e.js +1 -0
  272. mage_ai/server/{frontend_dist/_next/static/chunks/7361-694e1e4fb9c97d68.js → frontend_dist_base_path_template/_next/static/chunks/7361-6c5c9063b9f91700.js} +1 -1
  273. mage_ai/server/{frontend_dist/_next/static/chunks/7858-c83d25349d0e980f.js → frontend_dist_base_path_template/_next/static/chunks/7858-d9df72e95e438284.js} +1 -1
  274. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/{8146-92e7ccfed169ee9c.js → 8146-27f0e31f309897a5.js} +1 -1
  275. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/8264-0d582a6ca33c3dfa.js +1 -0
  276. mage_ai/server/{frontend_dist/_next/static/chunks/3684-cbda680fd8927d07.js → frontend_dist_base_path_template/_next/static/chunks/8432-f191e39f9b5893f2.js} +1 -1
  277. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/845-9a73c65fe3fdc328.js +1 -0
  278. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/8487-8e1c09546dff4dbf.js +1 -0
  279. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/90-a7308bae028d7001.js +1 -0
  280. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/9264-cc44b07f248707b0.js +1 -0
  281. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/9618-4eb49cdbd1ba11d7.js +1 -0
  282. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/{976-18c98af60b76f1a7.js → 976-0a8c2c4d7acd957b.js} +1 -1
  283. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/framework-22b71764bf44ede4.js +1 -0
  284. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/_app-78c4a077a2f279c2.js +1 -0
  285. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/block-layout-a24cb24b6f08bbc9.js +1 -0
  286. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/compute-419775ca1293b354.js +1 -0
  287. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/files-b37d221eb5ddc248.js +1 -0
  288. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/global-data-products/{[...slug]-3f6fc312f67ff72e.js → [...slug]-cfd68e760ae00958.js} +1 -1
  289. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/{global-data-products-f4cd03036c3e8723.js → global-data-products-c3b79ef31007f95b.js} +1 -1
  290. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/global-hooks/[...slug]-7adc543fc490367a.js +1 -0
  291. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/global-hooks-51d366993f6dd449.js +1 -0
  292. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/manage/files-fe6e73463a20d67c.js +1 -0
  293. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/manage/settings-60845f0b59142f32.js +1 -0
  294. mage_ai/server/{frontend_dist/_next/static/chunks/pages/manage/users/[user]-1d9b298fdceabbf1.js → frontend_dist_base_path_template/_next/static/chunks/pages/manage/users/[user]-9384c5f1efa2ac18.js} +1 -1
  295. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/manage/users/new-abd8571907664fdf.js +1 -0
  296. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/manage/{users-a7c970122a10afdc.js → users-28a930b148d99766.js} +1 -1
  297. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/manage-d8a38b5d1f50e798.js +1 -0
  298. mage_ai/server/{frontend_dist/_next/static/chunks/pages/overview-a4647a274cf4dffa.js → frontend_dist_base_path_template/_next/static/chunks/pages/overview-f0c40645f385f23f.js} +1 -1
  299. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/{pipeline-runs-c79819d6826e5416.js → pipeline-runs-b35d37bfba8fbccc.js} +1 -1
  300. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/backfills/[...slug]-9adc2974aada27ba.js +1 -0
  301. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/backfills-51b1311dff2a974e.js +1 -0
  302. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/dashboard-571c0962333b92f0.js +1 -0
  303. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/edit-2333e524d34b474a.js +1 -0
  304. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/{logs-a52d2d3e0c2978f4.js → logs-ef680455ae54ccbe.js} +1 -1
  305. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/monitors/block-runs-ddddcddd2f74b4f6.js +1 -0
  306. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/monitors/block-runtime-4a238307feddb522.js +1 -0
  307. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/monitors-6d1afeb4a84f50f7.js +1 -0
  308. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/runs/[run]-1ff9bb8e22ca1e75.js +1 -0
  309. mage_ai/server/{frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/runs-ce717786f31e8f04.js → frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/runs-2d20b2cd08907afd.js} +1 -1
  310. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/settings-11363aa58d51f4e1.js +1 -0
  311. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/syncs-b75bf17498e87354.js +1 -0
  312. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/triggers/[...slug]-779c3ef0676a12ac.js +1 -0
  313. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/triggers-bbea9a088657404a.js +1 -0
  314. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/{[pipeline]-35fe7762cb83a733.js → [pipeline]-02c843b9c8418bb5.js} +1 -1
  315. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines-3737f2b0afc2ede3.js +1 -0
  316. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/account/profile-3c8f062913c66f3e.js +1 -0
  317. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/permissions/{[...slug]-d6a62284c7c99cb3.js → [...slug]-b78b1be5b9ed84b9.js} +1 -1
  318. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/{permissions-b1389695f758c32b.js → permissions-37b78a436eeab258.js} +1 -1
  319. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/preferences-c1f4ed17d501ccca.js +1 -0
  320. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/roles/{[...slug]-0bdb66265286ab22.js → [...slug]-db05a80d18c168e5.js} +1 -1
  321. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/{roles-e04e18fc295ca76a.js → roles-f55c77e4f46c8d33.js} +1 -1
  322. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/sync-data-2a1f8737561fdd94.js +1 -0
  323. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/users/{[...slug]-0267358c91122109.js → [...slug]-e3bf6e5d8bb250c4.js} +1 -1
  324. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/{users-50fed18bb9b8d142.js → users-20f0a050a42a015d.js} +1 -1
  325. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/{settings-56f83205752b1323.js → settings-0f0121db7f5ff93d.js} +1 -1
  326. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/sign-in-99e2748e3c1d57a3.js +1 -0
  327. mage_ai/server/{frontend_dist/_next/static/chunks/pages/templates/[...slug]-9370551ffa462144.js → frontend_dist_base_path_template/_next/static/chunks/pages/templates/[...slug]-f44ccd1499ffd23a.js} +1 -1
  328. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/{templates-3cff90df13a1a755.js → templates-1bfaa1c50e844813.js} +1 -1
  329. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/terminal-ed121e305169cf1c.js +1 -0
  330. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/test-9ae68758102cc843.js +1 -0
  331. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/triggers-8bdd858240d5dbf6.js +1 -0
  332. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/version-control-dd4fb405695f74bf.js +1 -0
  333. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/{webpack-074f3eb2281fba79.js → webpack-d30cb09c85b4c4f0.js} +1 -1
  334. mage_ai/server/frontend_dist_base_path_template/block-layout.html +2 -2
  335. mage_ai/server/frontend_dist_base_path_template/compute.html +5 -5
  336. mage_ai/server/frontend_dist_base_path_template/files.html +5 -5
  337. mage_ai/server/frontend_dist_base_path_template/global-data-products/[...slug].html +5 -5
  338. mage_ai/server/frontend_dist_base_path_template/global-data-products.html +5 -5
  339. mage_ai/server/frontend_dist_base_path_template/global-hooks/[...slug].html +24 -0
  340. mage_ai/server/frontend_dist_base_path_template/global-hooks.html +24 -0
  341. mage_ai/server/frontend_dist_base_path_template/index.html +2 -2
  342. mage_ai/server/frontend_dist_base_path_template/manage/files.html +5 -5
  343. mage_ai/server/frontend_dist_base_path_template/manage/settings.html +5 -5
  344. mage_ai/server/frontend_dist_base_path_template/manage/users/[user].html +5 -5
  345. mage_ai/server/frontend_dist_base_path_template/manage/users/new.html +5 -5
  346. mage_ai/server/frontend_dist_base_path_template/manage/users.html +5 -5
  347. mage_ai/server/frontend_dist_base_path_template/manage.html +5 -5
  348. mage_ai/server/frontend_dist_base_path_template/oauth.html +4 -4
  349. mage_ai/server/frontend_dist_base_path_template/overview.html +5 -5
  350. mage_ai/server/frontend_dist_base_path_template/pipeline-runs.html +5 -5
  351. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/backfills/[...slug].html +5 -5
  352. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/backfills.html +5 -5
  353. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/dashboard.html +5 -5
  354. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/edit.html +2 -2
  355. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/logs.html +5 -5
  356. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/monitors/block-runs.html +5 -5
  357. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/monitors/block-runtime.html +5 -5
  358. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/monitors.html +5 -5
  359. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/runs/[run].html +5 -5
  360. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/runs.html +5 -5
  361. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/settings.html +5 -5
  362. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/syncs.html +5 -5
  363. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/triggers/[...slug].html +5 -5
  364. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline]/triggers.html +5 -5
  365. mage_ai/server/frontend_dist_base_path_template/pipelines/[pipeline].html +2 -2
  366. mage_ai/server/frontend_dist_base_path_template/pipelines.html +5 -5
  367. mage_ai/server/frontend_dist_base_path_template/settings/account/profile.html +5 -5
  368. mage_ai/server/frontend_dist_base_path_template/settings/workspace/permissions/[...slug].html +5 -5
  369. mage_ai/server/frontend_dist_base_path_template/settings/workspace/permissions.html +5 -5
  370. mage_ai/server/frontend_dist_base_path_template/settings/workspace/preferences.html +5 -5
  371. mage_ai/server/frontend_dist_base_path_template/settings/workspace/roles/[...slug].html +5 -5
  372. mage_ai/server/frontend_dist_base_path_template/settings/workspace/roles.html +5 -5
  373. mage_ai/server/frontend_dist_base_path_template/settings/workspace/sync-data.html +5 -5
  374. mage_ai/server/frontend_dist_base_path_template/settings/workspace/users/[...slug].html +5 -5
  375. mage_ai/server/frontend_dist_base_path_template/settings/workspace/users.html +5 -5
  376. mage_ai/server/frontend_dist_base_path_template/settings.html +2 -2
  377. mage_ai/server/frontend_dist_base_path_template/sign-in.html +28 -29
  378. mage_ai/server/frontend_dist_base_path_template/templates/[...slug].html +5 -5
  379. mage_ai/server/frontend_dist_base_path_template/templates.html +5 -5
  380. mage_ai/server/frontend_dist_base_path_template/terminal.html +5 -5
  381. mage_ai/server/frontend_dist_base_path_template/test.html +15 -15
  382. mage_ai/server/frontend_dist_base_path_template/triggers.html +5 -5
  383. mage_ai/server/frontend_dist_base_path_template/version-control.html +5 -5
  384. mage_ai/server/server.py +44 -12
  385. mage_ai/server/utils/output_display.py +6 -0
  386. mage_ai/server/websocket_server.py +14 -0
  387. mage_ai/services/aws/emr/config.py +1 -0
  388. mage_ai/services/aws/emr/constants.py +1 -0
  389. mage_ai/services/aws/emr/emr.py +20 -7
  390. mage_ai/services/compute/__init__.py +0 -0
  391. mage_ai/services/compute/aws/__init__.py +0 -0
  392. mage_ai/services/compute/aws/constants.py +21 -0
  393. mage_ai/services/compute/aws/models.py +459 -0
  394. mage_ai/services/compute/aws/steps.py +482 -0
  395. mage_ai/services/compute/constants.py +27 -0
  396. mage_ai/services/compute/models.py +212 -0
  397. mage_ai/services/k8s/job_manager.py +4 -0
  398. mage_ai/services/spark/api/aws_emr.py +38 -0
  399. mage_ai/services/spark/api/base.py +7 -4
  400. mage_ai/services/spark/api/constants.py +4 -0
  401. mage_ai/services/spark/api/local.py +25 -24
  402. mage_ai/services/spark/api/service.py +15 -5
  403. mage_ai/services/spark/constants.py +1 -1
  404. mage_ai/services/spark/models/applications.py +45 -3
  405. mage_ai/services/spark/models/base.py +3 -19
  406. mage_ai/services/spark/models/environments.py +16 -11
  407. mage_ai/services/spark/models/executors.py +2 -2
  408. mage_ai/services/spark/models/sqls.py +46 -15
  409. mage_ai/services/spark/models/stages.py +55 -32
  410. mage_ai/services/spark/models/threads.py +2 -2
  411. mage_ai/services/spark/utils.py +22 -6
  412. mage_ai/services/ssh/__init__.py +0 -0
  413. mage_ai/services/ssh/aws/__init__.py +0 -0
  414. mage_ai/services/ssh/aws/emr/__init__.py +0 -0
  415. mage_ai/services/ssh/aws/emr/constants.py +10 -0
  416. mage_ai/services/ssh/aws/emr/models.py +326 -0
  417. mage_ai/services/ssh/aws/emr/utils.py +151 -0
  418. mage_ai/settings/__init__.py +8 -1
  419. mage_ai/settings/repo.py +3 -0
  420. mage_ai/settings/secret_generation.py +7 -0
  421. mage_ai/settings/sso.py +8 -0
  422. mage_ai/shared/files.py +47 -0
  423. mage_ai/shared/hash.py +16 -0
  424. mage_ai/shared/models.py +254 -0
  425. mage_ai/streaming/constants.py +3 -0
  426. mage_ai/streaming/sinks/postgres.py +2 -0
  427. mage_ai/streaming/sinks/rabbitmq.py +76 -0
  428. mage_ai/streaming/sinks/sink_factory.py +4 -0
  429. mage_ai/streaming/sources/activemq.py +89 -0
  430. mage_ai/streaming/sources/nats_js.py +182 -0
  431. mage_ai/streaming/sources/source_factory.py +8 -0
  432. mage_ai/tests/ai/test_ai_functions.py +53 -8
  433. mage_ai/tests/api/endpoints/test_oauths.py +33 -0
  434. mage_ai/tests/api/endpoints/test_projects.py +1 -0
  435. mage_ai/tests/api/endpoints/test_workspaces.py +55 -0
  436. mage_ai/tests/api/operations/test_base.py +7 -5
  437. mage_ai/tests/api/operations/test_operations.py +0 -1
  438. mage_ai/tests/api/operations/test_operations_with_hooks.py +622 -0
  439. mage_ai/tests/api/operations/test_syncs.py +0 -1
  440. mage_ai/tests/api/operations/test_users.py +13 -2
  441. mage_ai/tests/api/policies/test_oauth_policy.py +38 -0
  442. mage_ai/tests/data_preparation/models/global_hooks/__init__.py +0 -0
  443. mage_ai/tests/data_preparation/models/global_hooks/test_global_hooks.py +600 -0
  444. mage_ai/tests/data_preparation/models/global_hooks/test_hook.py +804 -0
  445. mage_ai/tests/data_preparation/models/global_hooks/test_predicates.py +803 -0
  446. mage_ai/tests/data_preparation/models/global_hooks/test_utils.py +38 -0
  447. mage_ai/tests/data_preparation/models/test_block.py +26 -0
  448. mage_ai/tests/data_preparation/models/test_pipeline.py +15 -0
  449. mage_ai/tests/data_preparation/test_repo_manager.py +11 -0
  450. mage_ai/tests/factory.py +105 -5
  451. mage_ai/tests/orchestration/queue/test_process_queue.py +15 -2
  452. mage_ai/tests/orchestration/test_pipeline_scheduler.py +82 -1
  453. mage_ai/tests/services/k8s/test_job_manager.py +38 -1
  454. mage_ai/tests/shared/mixins.py +328 -0
  455. mage_ai/tests/shared/test_hash.py +17 -1
  456. mage_ai/tests/streaming/sinks/test_rabbitmq.py +36 -0
  457. mage_ai/tests/streaming/sources/test_activemq.py +32 -0
  458. mage_ai/tests/streaming/sources/test_nats_js.py +32 -0
  459. mage_ai/tests/streaming/sources/test_source_factory.py +26 -1
  460. {mage_ai-0.9.45.dist-info → mage_ai-0.9.47.dist-info}/METADATA +14 -4
  461. {mage_ai-0.9.45.dist-info → mage_ai-0.9.47.dist-info}/RECORD +467 -386
  462. {mage_ai-0.9.45.dist-info → mage_ai-0.9.47.dist-info}/WHEEL +1 -1
  463. mage_ai/authentication/oauth/active_directory.py +0 -17
  464. mage_ai/server/frontend_dist/_next/static/chunks/1125-91d3ce33140ef041.js +0 -1
  465. mage_ai/server/frontend_dist/_next/static/chunks/1749-607014ecf28268bf.js +0 -1
  466. mage_ai/server/frontend_dist/_next/static/chunks/1952-573c7fc7ad84da6e.js +0 -1
  467. mage_ai/server/frontend_dist/_next/static/chunks/3004-231cad9039ae5dcb.js +0 -1
  468. mage_ai/server/frontend_dist/_next/static/chunks/3419-0873e170ef7d6303.js +0 -1
  469. mage_ai/server/frontend_dist/_next/static/chunks/3943-9e1105393a3be0de.js +0 -1
  470. mage_ai/server/frontend_dist/_next/static/chunks/4267-335766a915ee2fa9.js +0 -1
  471. mage_ai/server/frontend_dist/_next/static/chunks/5457-8be2e0a3fe0ba64b.js +0 -1
  472. mage_ai/server/frontend_dist/_next/static/chunks/5499-bca977f466e259e1.js +0 -1
  473. mage_ai/server/frontend_dist/_next/static/chunks/553-edf533e634e85192.js +0 -1
  474. mage_ai/server/frontend_dist/_next/static/chunks/5810-37c6091b29a1fe53.js +0 -1
  475. mage_ai/server/frontend_dist/_next/static/chunks/600-0733eb84f0a0a9e0.js +0 -1
  476. mage_ai/server/frontend_dist/_next/static/chunks/6333-95ad799d13326dce.js +0 -1
  477. mage_ai/server/frontend_dist/_next/static/chunks/722-2a171fe616310f60.js +0 -1
  478. mage_ai/server/frontend_dist/_next/static/chunks/8224-39a93ee1058b6069.js +0 -1
  479. mage_ai/server/frontend_dist/_next/static/chunks/8264-6ef8fdb195694807.js +0 -1
  480. mage_ai/server/frontend_dist/_next/static/chunks/8487-032ef9b17d20aad9.js +0 -1
  481. mage_ai/server/frontend_dist/_next/static/chunks/9264-1b5c4b071ed544c3.js +0 -1
  482. mage_ai/server/frontend_dist/_next/static/chunks/framework-7c365855dab1bf41.js +0 -1
  483. mage_ai/server/frontend_dist/_next/static/chunks/pages/_app-1203afde83fa2d6e.js +0 -1
  484. mage_ai/server/frontend_dist/_next/static/chunks/pages/block-layout-c465c14d48392b11.js +0 -1
  485. mage_ai/server/frontend_dist/_next/static/chunks/pages/compute-b1f8d5a7a9a30f2d.js +0 -1
  486. mage_ai/server/frontend_dist/_next/static/chunks/pages/files-93c094bad0f299fb.js +0 -1
  487. mage_ai/server/frontend_dist/_next/static/chunks/pages/manage/files-891e5bd5935f7473.js +0 -1
  488. mage_ai/server/frontend_dist/_next/static/chunks/pages/manage/settings-d2e4ee4e68d36807.js +0 -1
  489. mage_ai/server/frontend_dist/_next/static/chunks/pages/manage/users/new-4a49126fcfe8ced0.js +0 -1
  490. mage_ai/server/frontend_dist/_next/static/chunks/pages/manage-5f8c5d0bc6ad1113.js +0 -1
  491. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/backfills/[...slug]-b4c0a36c69c346a4.js +0 -1
  492. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/backfills-a9266d353f288e8c.js +0 -1
  493. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/dashboard-b5c29c852262312e.js +0 -1
  494. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/edit-9f233917aa72ad82.js +0 -1
  495. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/monitors/block-runs-5af599da035252a7.js +0 -1
  496. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/monitors/block-runtime-8597f8e13d80ee48.js +0 -1
  497. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/monitors-5e0d047a6c9cb1fc.js +0 -1
  498. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/runs/[run]-b0a508aa52914dad.js +0 -1
  499. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/settings-074c32397d341de9.js +0 -1
  500. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/syncs-50f5d8706ed0bc73.js +0 -1
  501. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/triggers/[...slug]-e151c1552fcb67bd.js +0 -1
  502. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines/[pipeline]/triggers-95394769a783c6c7.js +0 -1
  503. mage_ai/server/frontend_dist/_next/static/chunks/pages/pipelines-8336c4326f1e7d96.js +0 -1
  504. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/account/profile-01dd679e4a21e0d9.js +0 -1
  505. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/preferences-4d9051c073a9b2ff.js +0 -1
  506. mage_ai/server/frontend_dist/_next/static/chunks/pages/settings/workspace/sync-data-5a306deca6717862.js +0 -1
  507. mage_ai/server/frontend_dist/_next/static/chunks/pages/sign-in-0dbc5b11019e78a0.js +0 -1
  508. mage_ai/server/frontend_dist/_next/static/chunks/pages/terminal-578d862f3e56e6e6.js +0 -1
  509. mage_ai/server/frontend_dist/_next/static/chunks/pages/test-1c0588d685b909b9.js +0 -1
  510. mage_ai/server/frontend_dist/_next/static/chunks/pages/triggers-ffaab4c013e62ba1.js +0 -1
  511. mage_ai/server/frontend_dist/_next/static/chunks/pages/version-control-114000e84313994f.js +0 -1
  512. mage_ai/server/frontend_dist/_next/static/icIDjCezcfhiKL87kgeSY/_buildManifest.js +0 -1
  513. mage_ai/server/frontend_dist_base_path_template/_next/static/_O1kJQLGtZmoBMyWpQrea/_buildManifest.js +0 -1
  514. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/1125-91d3ce33140ef041.js +0 -1
  515. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/1749-607014ecf28268bf.js +0 -1
  516. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/1952-573c7fc7ad84da6e.js +0 -1
  517. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/3004-231cad9039ae5dcb.js +0 -1
  518. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/3419-0873e170ef7d6303.js +0 -1
  519. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/3943-9e1105393a3be0de.js +0 -1
  520. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/4267-335766a915ee2fa9.js +0 -1
  521. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/5457-8be2e0a3fe0ba64b.js +0 -1
  522. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/5499-bca977f466e259e1.js +0 -1
  523. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/553-edf533e634e85192.js +0 -1
  524. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/5810-37c6091b29a1fe53.js +0 -1
  525. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/600-0733eb84f0a0a9e0.js +0 -1
  526. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/6333-95ad799d13326dce.js +0 -1
  527. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/722-2a171fe616310f60.js +0 -1
  528. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/8224-39a93ee1058b6069.js +0 -1
  529. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/8264-6ef8fdb195694807.js +0 -1
  530. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/8487-032ef9b17d20aad9.js +0 -1
  531. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/9264-1b5c4b071ed544c3.js +0 -1
  532. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/framework-7c365855dab1bf41.js +0 -1
  533. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/_app-1203afde83fa2d6e.js +0 -1
  534. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/block-layout-c465c14d48392b11.js +0 -1
  535. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/compute-b1f8d5a7a9a30f2d.js +0 -1
  536. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/files-93c094bad0f299fb.js +0 -1
  537. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/manage/files-891e5bd5935f7473.js +0 -1
  538. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/manage/settings-d2e4ee4e68d36807.js +0 -1
  539. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/manage/users/new-4a49126fcfe8ced0.js +0 -1
  540. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/manage-5f8c5d0bc6ad1113.js +0 -1
  541. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/backfills/[...slug]-b4c0a36c69c346a4.js +0 -1
  542. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/backfills-a9266d353f288e8c.js +0 -1
  543. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/dashboard-b5c29c852262312e.js +0 -1
  544. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/edit-9f233917aa72ad82.js +0 -1
  545. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/monitors/block-runs-5af599da035252a7.js +0 -1
  546. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/monitors/block-runtime-8597f8e13d80ee48.js +0 -1
  547. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/monitors-5e0d047a6c9cb1fc.js +0 -1
  548. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/runs/[run]-b0a508aa52914dad.js +0 -1
  549. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/settings-074c32397d341de9.js +0 -1
  550. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/syncs-50f5d8706ed0bc73.js +0 -1
  551. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/triggers/[...slug]-e151c1552fcb67bd.js +0 -1
  552. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines/[pipeline]/triggers-95394769a783c6c7.js +0 -1
  553. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/pipelines-8336c4326f1e7d96.js +0 -1
  554. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/account/profile-01dd679e4a21e0d9.js +0 -1
  555. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/preferences-4d9051c073a9b2ff.js +0 -1
  556. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/settings/workspace/sync-data-5a306deca6717862.js +0 -1
  557. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/sign-in-0dbc5b11019e78a0.js +0 -1
  558. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/terminal-578d862f3e56e6e6.js +0 -1
  559. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/test-1c0588d685b909b9.js +0 -1
  560. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/triggers-ffaab4c013e62ba1.js +0 -1
  561. mage_ai/server/frontend_dist_base_path_template/_next/static/chunks/pages/version-control-114000e84313994f.js +0 -1
  562. /mage_ai/server/frontend_dist/_next/static/{icIDjCezcfhiKL87kgeSY → N3FS4bHv0jpYeeg672uYK}/_ssgManifest.js +0 -0
  563. /mage_ai/server/frontend_dist_base_path_template/_next/static/{_O1kJQLGtZmoBMyWpQrea → aoO6jYZLVlUGCCdY-wmy8}/_ssgManifest.js +0 -0
  564. {mage_ai-0.9.45.dist-info → mage_ai-0.9.47.dist-info}/LICENSE +0 -0
  565. {mage_ai-0.9.45.dist-info → mage_ai-0.9.47.dist-info}/entry_points.txt +0 -0
  566. {mage_ai-0.9.45.dist-info → mage_ai-0.9.47.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,803 @@
1
+ import json
2
+ import random
3
+ import uuid
4
+ from typing import Any, Dict, List, Tuple, Union
5
+
6
+ from mage_ai.api.resources.BaseResource import BaseResource
7
+ from mage_ai.api.resources.GenericResource import GenericResource
8
+ from mage_ai.data_preparation.models.block import Block
9
+ from mage_ai.data_preparation.models.constants import BlockType
10
+ from mage_ai.data_preparation.models.global_hooks.constants import (
11
+ PredicateAndOrOperator,
12
+ PredicateObjectType,
13
+ PredicateOperator,
14
+ PredicateValueDataType,
15
+ )
16
+ from mage_ai.data_preparation.models.global_hooks.predicates import (
17
+ HookPredicate,
18
+ PredicateValueType,
19
+ )
20
+ from mage_ai.data_preparation.models.pipeline import Pipeline
21
+ from mage_ai.shared.array import find
22
+ from mage_ai.shared.hash import dig, ignore_keys, merge_dict, set_value
23
+ from mage_ai.tests.base_test import AsyncDBTestCase
24
+
25
+ first_object_key = 'first_object_key'
26
+ model = dict(power=100)
27
+ data_type_to_value_mapping = {
28
+ PredicateValueDataType.BOOLEAN: True,
29
+ PredicateValueDataType.DICTIONARY: model,
30
+ PredicateValueDataType.FLOAT: 1.0,
31
+ PredicateValueDataType.INTEGER: 3,
32
+ PredicateValueDataType.LIST: [model],
33
+ PredicateValueDataType.STRING: json.dumps(model),
34
+ }
35
+
36
+
37
+ def build_operation_resource_settings(
38
+ right_value: Union[bool, Dict, float, int, List, str],
39
+ first_object_key=first_object_key,
40
+ repo_path: str = None,
41
+ ) -> Dict:
42
+ settings = {
43
+ BaseResource.__name__: dict(
44
+ right_object_keys=[first_object_key],
45
+ operation_resource=GenericResource({
46
+ first_object_key: right_value,
47
+ }, None),
48
+ ),
49
+ dict.__name__: dict(
50
+ right_object_keys=[first_object_key],
51
+ operation_resource={
52
+ first_object_key: right_value,
53
+ },
54
+ ),
55
+ list.__name__: dict(
56
+ right_object_keys=[0, first_object_key],
57
+ operation_resource=[
58
+ GenericResource({
59
+ first_object_key: right_value,
60
+ }, None),
61
+ ],
62
+ ),
63
+ Pipeline.__name__: dict(
64
+ right_object_keys=['variables', first_object_key],
65
+ operation_resource=Pipeline(uuid.uuid4().hex, config=dict(variables={
66
+ first_object_key: right_value,
67
+ })),
68
+ ),
69
+ }
70
+
71
+ if repo_path:
72
+ settings[Block.__name__] = dict(
73
+ right_object_keys=['configuration', first_object_key],
74
+ operation_resource=Block.create(
75
+ uuid.uuid4().hex,
76
+ BlockType.DATA_LOADER,
77
+ repo_path,
78
+ configuration={
79
+ first_object_key: right_value,
80
+ },
81
+ ),
82
+ )
83
+
84
+ return settings
85
+
86
+
87
+ class PredicatesTest(AsyncDBTestCase):
88
+ def test_validate_operator_includes(self):
89
+ value = uuid.uuid4().hex
90
+
91
+ for left_value_data_type, left_value, left_value_not_included in [
92
+ (
93
+ PredicateValueDataType.BOOLEAN,
94
+ True,
95
+ False,
96
+ ),
97
+ (
98
+ PredicateValueDataType.DICTIONARY,
99
+ dict(mage=value),
100
+ dict(mage=value, more=value),
101
+ ),
102
+ (
103
+ PredicateValueDataType.FLOAT,
104
+ 1.0,
105
+ 3.0,
106
+ ),
107
+ (
108
+ PredicateValueDataType.INTEGER,
109
+ 3,
110
+ 7,
111
+ ),
112
+ (
113
+ PredicateValueDataType.LIST,
114
+ [value],
115
+ [value, value],
116
+ ),
117
+ (
118
+ PredicateValueDataType.STRING,
119
+ value,
120
+ value + value,
121
+ ),
122
+ ]:
123
+ for left_value_use, assertion_func in [
124
+ (left_value, self.assertTrue),
125
+ (left_value_not_included, self.assertFalse),
126
+ ]:
127
+ predicate = HookPredicate.load(
128
+ right_value=[left_value],
129
+ right_value_type=PredicateValueType.load(
130
+ value_data_type=PredicateValueDataType.LIST,
131
+ value_type=PredicateValueType.load(
132
+ value_data_type=left_value_data_type,
133
+ ),
134
+ ),
135
+ operator=PredicateOperator.INCLUDES,
136
+ left_value=left_value_use,
137
+ left_value_type=PredicateValueType.load(
138
+ value_data_type=left_value_data_type,
139
+ ),
140
+ )
141
+
142
+ assertion_func(predicate.validate(None))
143
+
144
+ def test_validate_using_object_types(self):
145
+ value = uuid.uuid4().hex
146
+
147
+ right_object_keys = ['fire']
148
+ right_value_type = PredicateValueType.load(
149
+ value_data_type=PredicateValueDataType.STRING,
150
+ )
151
+ right_object = set_value({}, right_object_keys, value)
152
+
153
+ operator = PredicateOperator.EQUALS
154
+ left_object_keys = ['wind', 'level']
155
+ left_value_type = right_value_type
156
+ left_object = set_value({}, left_object_keys, value)
157
+
158
+ object_types = [
159
+ PredicateObjectType.ERROR,
160
+ PredicateObjectType.HOOK,
161
+ PredicateObjectType.META,
162
+ PredicateObjectType.METADATA,
163
+ PredicateObjectType.PAYLOAD,
164
+ PredicateObjectType.QUERY,
165
+ PredicateObjectType.RESOURCE,
166
+ PredicateObjectType.RESOURCE_ID,
167
+ PredicateObjectType.RESOURCE_PARENT_ID,
168
+ PredicateObjectType.USER,
169
+ ]
170
+ for right_object_type in object_types:
171
+ left_object_type = find(
172
+ lambda x, right_object_type=right_object_type: x != right_object_type,
173
+ object_types,
174
+ )
175
+
176
+ for left_value_use, assertion_func in [
177
+ (value, self.assertTrue),
178
+ (value + value, self.assertFalse),
179
+ ]:
180
+ left_object_use = set_value(
181
+ left_object,
182
+ left_object_keys,
183
+ left_value_use,
184
+ )
185
+
186
+ right_object_keys_use = right_object_keys
187
+ right_object_use = right_object
188
+ left_object_keys_use = left_object_keys
189
+ left_object_use = left_object
190
+
191
+ if right_object_type in [
192
+ PredicateObjectType.RESOURCE_ID,
193
+ PredicateObjectType.RESOURCE_PARENT_ID,
194
+ ]:
195
+ right_object_use = dig(right_object_use, right_object_keys)
196
+ right_object_keys_use = None
197
+
198
+ if left_object_type in [
199
+ PredicateObjectType.RESOURCE_ID,
200
+ PredicateObjectType.RESOURCE_PARENT_ID,
201
+ ]:
202
+ left_object_use = dig(left_object_use, left_object_keys)
203
+ left_object_keys_use = None
204
+
205
+ predicate = HookPredicate.load(
206
+ right_object_keys=right_object_keys_use,
207
+ right_object_type=right_object_type,
208
+ right_value_type=right_value_type,
209
+ operator=operator,
210
+ left_object_keys=left_object_keys_use,
211
+ left_object_type=left_object_type,
212
+ left_value_type=left_value_type,
213
+ )
214
+
215
+ settings = {
216
+ right_object_type.value: right_object_use,
217
+ left_object_type.value: left_object_use,
218
+ }
219
+
220
+ assertion_func(predicate.validate(None, **settings))
221
+
222
+ def test_validate_using_operation_resource(self):
223
+ right_value = uuid.uuid4().hex
224
+
225
+ for operation_resource_class_name, settings in build_operation_resource_settings(
226
+ right_value,
227
+ repo_path=self.repo_path,
228
+ ).items():
229
+ right_object_keys = settings['right_object_keys']
230
+ operation_resource = settings['operation_resource']
231
+
232
+ for left_value_use, assertion_func in [
233
+ (right_value, self.assertTrue),
234
+ (right_value + right_value, self.assertFalse),
235
+ ]:
236
+ right_object_keys_use = right_object_keys
237
+ if list.__name__ == operation_resource_class_name:
238
+ right_object_keys_use = right_object_keys[1:]
239
+
240
+ predicate = HookPredicate.load(
241
+ right_object_keys=right_object_keys_use,
242
+ right_object_type=PredicateObjectType.OPERATION_RESOURCE,
243
+ right_value_type=PredicateValueType.load(
244
+ value_data_type=PredicateValueDataType.STRING,
245
+ ),
246
+ operator=PredicateOperator.EQUALS,
247
+ left_value=left_value_use,
248
+ left_value_type=PredicateValueType.load(
249
+ value_data_type=PredicateValueDataType.STRING,
250
+ ),
251
+ )
252
+
253
+ assertion_func(predicate.validate(operation_resource))
254
+
255
+ def test_validate_multiple_operation_resources(self):
256
+ value = uuid.uuid4().hex
257
+ resource = dict(fire=value)
258
+
259
+ operation_resources_groups = [
260
+ (
261
+ self.assertTrue,
262
+ [resource, resource],
263
+ ),
264
+ (
265
+ self.assertFalse,
266
+ [resource, dict(fire=value + value)],
267
+ ),
268
+ ]
269
+
270
+ for assertion_func, operation_resources in operation_resources_groups:
271
+ predicate = HookPredicate.load(
272
+ right_object_keys=['fire'],
273
+ right_object_type=PredicateObjectType.OPERATION_RESOURCE,
274
+ right_value_type=PredicateValueType.load(
275
+ value_data_type=PredicateValueDataType.STRING,
276
+ ),
277
+ operator=PredicateOperator.EQUALS,
278
+ left_value=value,
279
+ left_value_type=PredicateValueType.load(
280
+ value_data_type=PredicateValueDataType.STRING,
281
+ ),
282
+ )
283
+
284
+ assertion_func(predicate.validate(operation_resources))
285
+
286
+ def test_validate_with_nested_predicates(self):
287
+ (
288
+ operation_resource,
289
+ predicate,
290
+ predicates_level_1,
291
+ predicates_level_2,
292
+ predicates_level_3,
293
+ ) = build_nested_predicates()
294
+
295
+ for pred_dicts in [
296
+ predicates_level_1,
297
+ predicates_level_2,
298
+ predicates_level_3,
299
+ ]:
300
+ for key, arr in pred_dicts.items():
301
+ for predicate_inner in arr:
302
+ assertion_func = self.assertTrue if 'succeed' == key else self.assertFalse
303
+
304
+ for predicate_inner in arr:
305
+ assertion_func(predicate_inner.validate(operation_resource))
306
+
307
+ self.assertTrue(predicate.validate(operation_resource))
308
+
309
+ def test_validate_with_nested_predicates_other_combinations(self):
310
+ (
311
+ operation_resource,
312
+ predicate,
313
+ predicates_level_1,
314
+ predicates_level_2,
315
+ predicates_level_3,
316
+ ) = build_nested_predicates()
317
+
318
+ predicates_failed = []
319
+ predicates_succeed = [predicate]
320
+
321
+ for pred_dict in [
322
+ predicates_level_1,
323
+ predicates_level_2,
324
+ predicates_level_3,
325
+ ]:
326
+ predicates_failed.extend(pred_dict['failed'])
327
+ predicates_succeed.extend(pred_dict['succeed'])
328
+
329
+ predicates_failed_count = len(predicates_failed)
330
+ predicates_succeed_count = len(predicates_succeed)
331
+ predicates_count = predicates_failed_count + predicates_succeed_count
332
+
333
+ def _get_and_or_operator() -> PredicateAndOrOperator:
334
+ return random.choice([v for v in PredicateAndOrOperator])
335
+
336
+ def _get_predicates(
337
+ operator: PredicateAndOrOperator,
338
+ force_failure: bool = False,
339
+ predicates_failed=predicates_failed,
340
+ predicates_succeed=predicates_succeed,
341
+ predicates_succeed_count=predicates_succeed_count,
342
+ ) -> List[HookPredicate]:
343
+ failed_sample = 0
344
+ succeed_sample = random.randint(1, predicates_succeed_count)
345
+
346
+ if PredicateAndOrOperator.OR == operator or force_failure:
347
+ failed_sample = random.randint(1, predicates_failed_count)
348
+
349
+ arr_failed = random.sample(predicates_failed, failed_sample)
350
+ arr_succeed = random.sample(predicates_succeed, succeed_sample)
351
+
352
+ if force_failure:
353
+ return arr_failed
354
+
355
+ return arr_failed + arr_succeed
356
+
357
+ levels = 10
358
+
359
+ def _build_all(
360
+ # force_failure_at_level: int = None,
361
+ levels=levels,
362
+ predicates_count=predicates_count,
363
+ get_and_or_operator=_get_and_or_operator,
364
+ get_predicates=_get_predicates,
365
+ ) -> List[List[HookPredicate]]:
366
+ predicates_in_reverse_level_order = []
367
+ for i1 in range(levels):
368
+ row = []
369
+ predicates_to_create_at_this_level = random.randint(
370
+ predicates_count - i1, predicates_count,
371
+ )
372
+ for _i2 in range(predicates_to_create_at_this_level):
373
+ operator = get_and_or_operator()
374
+ predicates = get_predicates(
375
+ operator,
376
+ # force_failure=(
377
+ # force_failure_at_level is not None and
378
+ # force_failure_at_level == levels - i1
379
+ # ),
380
+ )
381
+
382
+ pred = HookPredicate.load(
383
+ and_or_operator=operator,
384
+ predicates=predicates,
385
+ )
386
+ row.append(pred)
387
+
388
+ if i1 >= 1:
389
+ previous_row = predicates_in_reverse_level_order[i1 - 1]
390
+ batch_count = round(len(previous_row) / len(row))
391
+
392
+ for idx, pred in enumerate(row):
393
+ start_index = idx * batch_count
394
+ end_index = (idx + 1) * batch_count
395
+ pred.predicates += previous_row[start_index:end_index]
396
+
397
+ predicates_in_reverse_level_order.append(row)
398
+
399
+ return predicates_in_reverse_level_order
400
+
401
+ all_succeed = _build_all()
402
+
403
+ for i in range(levels):
404
+ level_to_run = levels - (i + 1)
405
+ for predicate0 in all_succeed[level_to_run]:
406
+ self.assertTrue(predicate0.validate(operation_resource))
407
+
408
+
409
+ class CustomTestError(Exception):
410
+ pass
411
+
412
+
413
+ def get_right_left_value(
414
+ right_value: Union[bool, Dict, float, int, List, str],
415
+ data_type: PredicateValueDataType,
416
+ object_type: PredicateObjectType,
417
+ operator: PredicateOperator,
418
+ ) -> Tuple[Union[bool, Dict, float, int, List, str], Union[bool, Dict, float, int, List, str]]:
419
+ if PredicateValueDataType.BOOLEAN == data_type:
420
+ if PredicateOperator.EQUALS == operator:
421
+ return (True, True)
422
+ elif PredicateOperator.GREATER_THAN == operator:
423
+ return (False, True)
424
+ elif PredicateOperator.INCLUDES == operator:
425
+ raise CustomTestError
426
+ elif PredicateOperator.LESS_THAN == operator:
427
+ return (True, False)
428
+ elif PredicateOperator.NOT_EQUALS == operator:
429
+ return (right_value, not right_value)
430
+ elif PredicateOperator.NOT_INCLUDES == operator:
431
+ raise CustomTestError
432
+ elif PredicateOperator.NOT_PRESENT == operator:
433
+ return (right_value, '')
434
+
435
+ if PredicateValueDataType.DICTIONARY == data_type:
436
+ if operator in [
437
+ PredicateOperator.INCLUDES,
438
+ PredicateOperator.NOT_INCLUDES,
439
+ ]:
440
+ raise CustomTestError
441
+
442
+ if PredicateOperator.NOT_EQUALS == operator:
443
+ return [right_value, merge_dict(right_value, dict(
444
+ random=uuid.uuid4().hex,
445
+ ))]
446
+
447
+ if PredicateOperator.GREATER_THAN == operator:
448
+ return (right_value, merge_dict(right_value, dict(
449
+ random=uuid.uuid4().hex,
450
+ )))
451
+
452
+ if PredicateOperator.LESS_THAN == operator:
453
+ return (merge_dict(right_value, dict(
454
+ random=uuid.uuid4().hex,
455
+ )), right_value)
456
+
457
+ if PredicateOperator.NOT_PRESENT == operator:
458
+ return (right_value, None)
459
+
460
+ if data_type in [PredicateValueDataType.FLOAT, PredicateValueDataType.INTEGER]:
461
+ if PredicateOperator.GREATER_THAN == operator:
462
+ return (right_value, right_value + 1)
463
+ elif PredicateOperator.INCLUDES == operator:
464
+ raise CustomTestError
465
+ elif PredicateOperator.LESS_THAN == operator:
466
+ return (right_value + 1, right_value)
467
+ elif PredicateOperator.NOT_EQUALS == operator:
468
+ return (right_value, right_value - right_value)
469
+ elif PredicateOperator.NOT_INCLUDES == operator:
470
+ raise CustomTestError
471
+ elif PredicateOperator.NOT_PRESENT == operator:
472
+ return (right_value, None)
473
+
474
+ if PredicateValueDataType.LIST == data_type:
475
+ if operator in [
476
+ PredicateOperator.INCLUDES,
477
+ PredicateOperator.NOT_INCLUDES,
478
+ ]:
479
+ raise CustomTestError
480
+
481
+ if PredicateOperator.NOT_EQUALS == operator:
482
+ return (right_value, right_value + right_value)
483
+
484
+ if PredicateOperator.GREATER_THAN == operator:
485
+ return (right_value, right_value + right_value)
486
+
487
+ if PredicateOperator.LESS_THAN == operator:
488
+ return (right_value + right_value, right_value)
489
+
490
+ if PredicateOperator.NOT_PRESENT == operator:
491
+ return (right_value, None)
492
+
493
+ if PredicateValueDataType.STRING == data_type:
494
+ if PredicateOperator.NOT_EQUALS == operator:
495
+ return (right_value, right_value + right_value)
496
+
497
+ if PredicateOperator.GREATER_THAN == operator:
498
+ return ('A', 'a')
499
+
500
+ if PredicateOperator.LESS_THAN == operator:
501
+ return ('a', 'A')
502
+
503
+ if PredicateOperator.NOT_INCLUDES == operator:
504
+ return (right_value, right_value + right_value)
505
+
506
+ if PredicateOperator.NOT_PRESENT == operator:
507
+ return (right_value, None)
508
+
509
+ return (right_value, right_value)
510
+
511
+
512
+ def build_test_validate(
513
+ data_type: PredicateValueDataType,
514
+ right_value: Any,
515
+ object_type: PredicateObjectType,
516
+ operator: PredicateOperator,
517
+ left_value: Any,
518
+ settings: Dict,
519
+ ):
520
+ async def _test_validate(
521
+ self,
522
+ data_type=data_type,
523
+ right_value=right_value,
524
+ object_type=object_type,
525
+ operator=operator,
526
+ left_value=left_value,
527
+ settings=settings,
528
+ ):
529
+ predicate = HookPredicate.load(
530
+ right_object_keys=settings.get('right_object_keys'),
531
+ right_object_type=object_type,
532
+ right_value=right_value,
533
+ right_value_type=PredicateValueType.load(
534
+ value_data_type=data_type,
535
+ ),
536
+ operator=operator,
537
+ left_value=left_value,
538
+ )
539
+
540
+ options = ignore_keys(settings, [
541
+ 'right_object_keys',
542
+ 'operation_resource',
543
+ ])
544
+
545
+ self.assertTrue(predicate.validate(
546
+ settings['operation_resource'],
547
+ **options,
548
+ ))
549
+
550
+ return _test_validate
551
+
552
+
553
+ for object_type in PredicateObjectType:
554
+ for data_type in PredicateValueDataType:
555
+ for operator in PredicateOperator:
556
+ right_value = data_type_to_value_mapping[data_type]
557
+ left_value = None
558
+
559
+ try:
560
+ right_value, left_value = get_right_left_value(
561
+ right_value,
562
+ data_type=data_type,
563
+ object_type=object_type,
564
+ operator=operator,
565
+ )
566
+ except CustomTestError:
567
+ continue
568
+
569
+ operation_resource_settings = {}
570
+ for tup in build_operation_resource_settings(right_value).items():
571
+ operation_resource_class_name, settings_init = tup
572
+ settings = settings_init.copy()
573
+
574
+ if object_type in [
575
+ PredicateObjectType.RESOURCE_ID,
576
+ PredicateObjectType.RESOURCE_PARENT_ID,
577
+ ]:
578
+ settings.update({
579
+ 'right_object_keys': None,
580
+ object_type.value: right_value,
581
+ })
582
+ elif PredicateObjectType.OPERATION_RESOURCE != object_type:
583
+ settings.update({
584
+ 'right_object_keys': [first_object_key],
585
+ object_type.value: {
586
+ first_object_key: right_value,
587
+ },
588
+ })
589
+ elif list.__name__ == operation_resource_class_name and \
590
+ PredicateObjectType.OPERATION_RESOURCE == object_type:
591
+
592
+ settings.update(dict(right_object_keys=[first_object_key]))
593
+
594
+ operation_resource_settings[operation_resource_class_name] = settings
595
+
596
+ for operation_resource_class_name, settings in operation_resource_settings.items():
597
+ method_name = '_'.join([
598
+ 'test_validate',
599
+ 'object_type',
600
+ object_type.value,
601
+ 'data_type',
602
+ data_type.value,
603
+ 'operator',
604
+ operator.value,
605
+ 'operation_resource',
606
+ operation_resource_class_name,
607
+ ])
608
+ setattr(
609
+ PredicatesTest,
610
+ method_name,
611
+ build_test_validate(
612
+ data_type=data_type,
613
+ right_value=right_value,
614
+ object_type=object_type,
615
+ operator=operator,
616
+ left_value=left_value,
617
+ settings=settings,
618
+ ),
619
+ )
620
+
621
+
622
+ def build_nested_predicates() -> Tuple[Dict, HookPredicate, Dict, Dict, Dict]:
623
+ operation_resource = dict(
624
+ earth=1,
625
+ fire=2,
626
+ ice=[3],
627
+ laser=4,
628
+ lightning=5,
629
+ power=6,
630
+ water=7,
631
+ wind=8,
632
+ )
633
+
634
+ def _build_settings(**kwargs) -> Dict:
635
+ right_value_type = PredicateValueType.load(
636
+ value_data_type=PredicateValueDataType.INTEGER,
637
+ )
638
+
639
+ return merge_dict(dict(
640
+ right_object_type=PredicateObjectType.OPERATION_RESOURCE,
641
+ right_value_type=right_value_type,
642
+ operator=PredicateOperator.EQUALS,
643
+ left_value_type=right_value_type,
644
+ ), kwargs)
645
+
646
+ predicate2_3_or_or_terminal_failed = HookPredicate.load(**_build_settings(
647
+ right_object_type=None,
648
+ right_value_type=None,
649
+ operator=PredicateOperator.PRESENT,
650
+ left_value=None,
651
+ left_value_type=None,
652
+ ))
653
+
654
+ predicate2_3_and_and_and_terminal_failed = HookPredicate.load(**_build_settings(
655
+ and_or_operator=PredicateAndOrOperator.AND,
656
+ right_object_keys=['ice'],
657
+ right_value_type=PredicateValueType.load(
658
+ value_data_type=PredicateValueDataType.LIST,
659
+ value_type=PredicateValueType.load(
660
+ value_data_type=PredicateValueDataType.INTEGER,
661
+ ),
662
+ ),
663
+ operator=PredicateOperator.NOT_INCLUDES,
664
+ left_value=3,
665
+ ))
666
+
667
+ predicate2_3_or_or_terminal_succeed = HookPredicate.load(**_build_settings(
668
+ right_object_keys=['power'],
669
+ operator=PredicateOperator.GREATER_THAN_OR_EQUALS,
670
+ left_value=6
671
+ ))
672
+
673
+ predicate2_3_and_and_or_terminal_succeed = HookPredicate.load(**_build_settings(
674
+ right_object_keys=['water'],
675
+ operator=PredicateOperator.LESS_THAN_OR_EQUALS,
676
+ left_value=6,
677
+ ))
678
+
679
+ predicate2_and_and_succeed = HookPredicate.load(**_build_settings(
680
+ and_or_operator=PredicateAndOrOperator.AND,
681
+ right_object_keys=['wind'],
682
+ operator=PredicateOperator.LESS_THAN_OR_EQUALS,
683
+ predicates=[
684
+ predicate2_3_and_and_or_terminal_succeed,
685
+ predicate2_3_or_or_terminal_succeed,
686
+ ],
687
+ left_value=8,
688
+ ))
689
+
690
+ predicate2_and_or_succeed = HookPredicate.load(**_build_settings(
691
+ and_or_operator=PredicateAndOrOperator.OR,
692
+ right_object_keys=['water'],
693
+ operator=PredicateOperator.NOT_EQUALS,
694
+ predicates=[
695
+ predicate2_3_and_and_or_terminal_succeed,
696
+ ],
697
+ left_value=8,
698
+ ))
699
+
700
+ predicate2_or_and_failed = HookPredicate.load(**_build_settings(
701
+ and_or_operator=PredicateAndOrOperator.AND,
702
+ right_object_keys=['ice'],
703
+ right_value_type=PredicateValueType.load(
704
+ value_data_type=PredicateValueDataType.LIST,
705
+ value_type=PredicateValueType.load(
706
+ value_data_type=PredicateValueDataType.INTEGER,
707
+ ),
708
+ ),
709
+ operator=PredicateOperator.INCLUDES,
710
+ predicates=[
711
+ predicate2_3_and_and_and_terminal_failed,
712
+ predicate2_3_and_and_or_terminal_succeed,
713
+ ],
714
+ left_value=3,
715
+ ))
716
+
717
+ predicate2_or_or_succeed = HookPredicate.load(**_build_settings(
718
+ and_or_operator=PredicateAndOrOperator.OR,
719
+ right_value_type=None,
720
+ operator=PredicateOperator.NOT_PRESENT,
721
+ predicates=[
722
+ predicate2_3_or_or_terminal_failed,
723
+ predicate2_3_or_or_terminal_succeed,
724
+ ],
725
+ left_value=None,
726
+ left_value_type=None,
727
+ ))
728
+
729
+ predicate1_and_succeed = HookPredicate.load(**_build_settings(
730
+ and_or_operator=PredicateAndOrOperator.AND,
731
+ right_object_keys=['ice'],
732
+ right_value_type=PredicateValueType.load(
733
+ value_data_type=PredicateValueDataType.LIST,
734
+ value_type=PredicateValueType.load(
735
+ value_data_type=PredicateValueDataType.INTEGER,
736
+ ),
737
+ ),
738
+ operator=PredicateOperator.INCLUDES,
739
+ predicates=[
740
+ predicate2_and_and_succeed,
741
+ predicate2_and_or_succeed,
742
+ ],
743
+ left_value=3,
744
+ ))
745
+
746
+ predicate1_or_succeed = HookPredicate.load(**_build_settings(
747
+ and_or_operator=PredicateAndOrOperator.OR,
748
+ right_object_keys=['fire'],
749
+ operator=PredicateOperator.GREATER_THAN,
750
+ predicates=[
751
+ predicate2_or_and_failed,
752
+ predicate2_or_or_succeed,
753
+ ],
754
+ left_value=3,
755
+ ))
756
+
757
+ predicates_level_3 = dict(
758
+ failed=[
759
+ predicate2_3_and_and_and_terminal_failed,
760
+ predicate2_3_or_or_terminal_failed,
761
+ ],
762
+ succeed=[
763
+ predicate2_3_and_and_or_terminal_succeed,
764
+ predicate2_3_or_or_terminal_succeed,
765
+ ],
766
+ )
767
+
768
+ predicates_level_2 = dict(
769
+ failed=[
770
+ predicate2_or_and_failed,
771
+ ],
772
+ succeed=[
773
+ predicate2_and_and_succeed,
774
+ predicate2_and_or_succeed,
775
+ predicate2_or_or_succeed,
776
+ ],
777
+ )
778
+
779
+ predicates_level_1 = dict(
780
+ failed=[],
781
+ succeed=[
782
+ predicate1_and_succeed,
783
+ predicate1_or_succeed,
784
+ ],
785
+ )
786
+
787
+ predicate = HookPredicate.load(**_build_settings(
788
+ and_or_operator=PredicateAndOrOperator.AND,
789
+ right_object_keys=['earth'],
790
+ predicates=[
791
+ predicate1_and_succeed,
792
+ predicate1_or_succeed,
793
+ ],
794
+ left_value=1,
795
+ ))
796
+
797
+ return (
798
+ operation_resource,
799
+ predicate,
800
+ predicates_level_1,
801
+ predicates_level_2,
802
+ predicates_level_3,
803
+ )