Feature/integrations pt1 integrations

src/secops/chronicle/__init__.py

@@ -98,27 +98,43 @@
     search_log_types,
 )
 from secops.chronicle.models import (
+    AdvancedConfig,
     AlertCount,
     AlertState,
     Case,
     CaseList,
+    DailyScheduleDetails,
     DataExport,
     DataExportStage,
     DataExportStatus,
+    Date,
+    DayOfWeek,
     DetectionType,
+    DiffType,
     Entity,
     EntityMetadata,
     EntityMetrics,
     EntitySummary,
     FileMetadataAndProperties,
     InputInterval,
+    IntegrationJobInstanceParameter,
+    IntegrationParam,
+    IntegrationParamType,
+    IntegrationType,
     ListBasis,
+    MonthlyScheduleDetails,
+    OneTimeScheduleDetails,
     PrevalenceData,
+    PythonVersion,
+    ScheduleType,
     SoarPlatformInfo,
+    TargetMode,
     TileType,
     TimeInterval,
     Timeline,
     TimelineBucket,
+    TimeOfDay,
+    WeeklyScheduleDetails,
     WidgetMetadata,
 )
 from secops.chronicle.nl_search import translate_nl_to_udm
@@ -198,6 +214,26 @@
     create_watchlist,
     update_watchlist,
 )
+from secops.chronicle.integration.integrations import (
+    list_integrations,
+    get_integration,
+    delete_integration,
+    create_integration,
+    transition_integration,
+    update_integration,
+    update_custom_integration,
+    get_integration_affected_items,
+    get_integration_dependencies,
+    get_integration_diff,
+    get_integration_restricted_agents,
+)
+from secops.chronicle.integration.marketplace_integrations import (
+    list_marketplace_integrations,
+    get_marketplace_integration,
+    get_marketplace_integration_diff,
+    install_marketplace_integration,
+    uninstall_marketplace_integration,
+)
 
 __all__ = [
     # Client
@@ -315,21 +351,31 @@
     "execute_query",
     "get_execute_query",
     # Models
+    "AdvancedConfig",
+    "AlertCount",
+    "AlertState",
+    "Case",
+    "CaseList",
+    "DailyScheduleDetails",
+    "Date",
+    "DayOfWeek",
     "Entity",
     "EntityMetadata",
     "EntityMetrics",
+    "EntitySummary",
+    "FileMetadataAndProperties",
+    "IntegrationJobInstanceParameter",
+    "MonthlyScheduleDetails",
+    "OneTimeScheduleDetails",
+    "PrevalenceData",
+    "ScheduleType",
+    "SoarPlatformInfo",
     "TimeInterval",
-    "TimelineBucket",
     "Timeline",
+    "TimelineBucket",
+    "TimeOfDay",
+    "WeeklyScheduleDetails",
     "WidgetMetadata",
-    "EntitySummary",
-    "AlertCount",
-    "AlertState",
-    "Case",
-    "SoarPlatformInfo",
-    "CaseList",
-    "PrevalenceData",
-    "FileMetadataAndProperties",
     "ValidationResult",
     "GeminiResponse",
     "Block",
@@ -367,4 +413,22 @@
     "delete_watchlist",
     "create_watchlist",
     "update_watchlist",
+    # Integrations
+    "list_integrations",
+    "get_integration",
+    "delete_integration",
+    "create_integration",
+    "transition_integration",
+    "update_integration",
+    "update_custom_integration",
+    "get_integration_affected_items",
+    "get_integration_dependencies",
+    "get_integration_diff",
+    "get_integration_restricted_agents",
+    # Marketplace Integrations
+    "list_marketplace_integrations",
+    "get_marketplace_integration",
+    "get_marketplace_integration_diff",
+    "install_marketplace_integration",
+    "uninstall_marketplace_integration",
 ]