dagster-cloud 1.8.2__py3-none-any.whl → 1.12.6__py3-none-any.whl

dagster_cloud/storage/event_logs/queries.py

@@ -105,6 +105,8 @@ ASSET_ENTRY_FRAGMENT = (
         lastObservationRecord {
             ...EventLogRecordFragment
         }
+        lastPlannedMaterializationStorageId
+        lastPlannedMaterializationRunId
     }
     """
 )
@@ -140,11 +142,28 @@ ASSET_CHECK_STATE_FRAGMENT = (
             }
             createTimestamp
         }
-        lastRunId 
-    } 
+        lastCompletedCheckExecutionRecord {
+            id
+            runId
+            status
+            event {
+                ...EventLogEntryFragment
+            }
+            createTimestamp
+        }
+        lastRunId
+    }
 """
 )
 
+GET_MAXIMUM_RECORD_ID = """
+    query getMaximumRecordId {
+        eventLogs {
+            getMaximumRecordId
+        }
+    }
+    """
+
 GET_RECORDS_FOR_RUN_QUERY = (
     EVENT_RECORD_FRAGMENT
     + """
@@ -321,9 +340,9 @@ GET_MATERIALIZATION_COUNT_BY_PARTITION = """
     """
 
 GET_LATEST_STORAGE_ID_BY_PARTITION = """
-    query getLatestStorageIdByPartition($assetKey: String!, $eventType: String!) {
+    query getLatestStorageIdByPartition($assetKey: String!, $eventType: String!, $partitions: [String!]) {
         eventLogs {
-            getLatestStorageIdByPartition(assetKey: $assetKey, eventType: $eventType) {
+            getLatestStorageIdByPartition(assetKey: $assetKey, eventType: $eventType, partitions: $partitions) {
                 ... on StorageIdByPartition {
                     partition
                     storageId
@@ -465,6 +484,16 @@ WIPE_ASSET_MUTATION = """
     }
 """
 
+WIPE_ASSET_PARTITIONS_MUTATION = """
+    mutation WipeAssetPartitions($assetKey: String!, $partitionKeys: [String!]!) {
+        eventLogs {
+            WipeAssetPartitions(assetKey: $assetKey, partitionKeys: $partitionKeys) {
+                ok
+            }
+        }
+    }
+"""
+
 WIPE_ASSET_CACHED_STATUS_DATA_MUTATION = """
     mutation WipeAssetCachedStatusData($assetKey: String!) {
         eventLogs {
@@ -483,6 +512,19 @@ query getDynamicPartitions($partitionsDefName: String!) {
 }
 """
 
+GET_PAGINATED_DYNAMIC_PARTITIONS_QUERY = """
+query getPaginatedDynamicPartitions($partitionsDefName: String!, $limit: Int!, $ascending: Boolean!, $cursor: String) {
+    eventLogs {
+        getPaginatedDynamicPartitions(partitionsDefName: $partitionsDefName, limit: $limit, ascending: $ascending, cursor: $cursor) {
+            results
+            hasMore
+            cursor
+        }
+    }
+}
+"""
+
+
 HAS_DYNAMIC_PARTITION_QUERY = """
 query hasDynamicPartition($partitionsDefName: String!, $partitionKey: String!) {
     eventLogs {
@@ -566,6 +608,30 @@ query getConcurrencyKeys {
 }
 """
 
+GET_POOL_LIMITS_QUERY = """
+query getPoolLimits {
+    eventLogs {
+        getPoolLimits {
+            name
+            limit
+            fromDefault
+        }
+    }
+}
+"""
+
+GET_POOL_CONFIG_QUERY = """
+query getPoolConfig {
+    eventLogs {
+        getPoolConfig {
+            poolGranularity
+            defaultPoolLimit
+            opGranularityRunBuffer
+        }
+    }
+}
+"""
+
 GET_CONCURRENCY_INFO_QUERY = """
 query getConcurrencyInfo($concurrencyKey: String!) {
     eventLogs {
@@ -583,6 +649,8 @@ query getConcurrencyInfo($concurrencyKey: String!) {
                 assignedTimestamp
                 priority
             }
+            limit
+            usingDefaultLimit
         }
     }
 }
@@ -705,6 +773,23 @@ query getRunStatusChangeEventRecordsQuery($recordsFilter: RunStatusChangeRecords
 """
 )
 
+FETCH_FAILED_MATERIALIZATIONS_QUERY = (
+    EVENT_RECORD_FRAGMENT
+    + """
+query fetchFailedMaterializationsQuery($recordsFilter: AssetRecordsFilter!, $limit: Int, $cursor: String, $ascending: Boolean) {
+    eventLogs {
+        fetchFailedMaterializations(recordsFilter: $recordsFilter, limit: $limit, cursor: $cursor, ascending: $ascending) {
+            records {
+                ...EventLogRecordFragment
+            }
+            cursor
+            hasMore
+        }
+    }
+}
+"""
+)
+
 GET_LATEST_PLANNED_MATERIALIZATION_INFO = """
     query getLatestPlannedMaterializationInfo($assetKey: String!, $partition: String) {
         eventLogs {
@@ -724,3 +809,16 @@ GET_UPDATED_DATA_VERSION_PARTITIONS = """
         }
     }
 """
+
+GET_ASSET_STATUS_CACHE_VALUES = (
+    CACHED_STATUS_DATA_FRAGMENT
+    + """
+    query getAssetStatusCacheValues($assetKeys: [AssetKeyInput!]!) {
+        eventLogs {
+            getAssetStatusCacheValues(assetKeys: $assetKeys) {
+                ...CachedStatusDataFragment
+            }
+        }
+    }
+"""
+)