feat(data-warehouse): Added vitally Source

frontend/src/scenes/data-warehouse/new/sourceWizardLogic.tsx

@@ -567,6 +567,45 @@ export const SOURCE_DETAILS: Record<ExternalDataSourceType, SourceConfig> = {
         ],
         caption: 'Select an existing Salesforce account to link to PostHog or create a new connection',
     },
+    Vitally: {
+        name: 'Vitally',
+        fields: [
+            {
+                name: 'secret_token',
+                label: 'Secret token',
+                type: 'text',
+                required: true,
+                placeholder: 'sk_live_...',
+            },
+            {
+                type: 'select',
+                name: 'region',
+                label: 'Vitally region',
+                required: true,
+                defaultValue: 'EU',
+                options: [
+                    {
+                        label: 'EU',
+                        value: 'EU',
+                    },
+                    {
+                        label: 'US',
+                        value: 'US',
+                        fields: [
+                            {
+                                name: 'subdomain',
+                                label: 'Vitally subdomain',
+                                type: 'text',
+                                required: true,
+                                placeholder: '',
+                            },
+                        ],
+                    },
+                ],
+            },
+        ],
+        caption: '',
+    },
 }
 
 export const buildKeaFormDefaultFromSourceDetails = (

frontend/src/scenes/data-warehouse/settings/DataWarehouseManagedSourcesTable.tsx

@@ -14,6 +14,7 @@ import IconSalesforce from 'public/services/salesforce.png'
 import IconSnowflake from 'public/services/snowflake.png'
 import IconMSSQL from 'public/services/sql-azure.png'
 import IconStripe from 'public/services/stripe.png'
+import IconVitally from 'public/services/vitally.png'
 import IconZendesk from 'public/services/zendesk.png'
 import { urls } from 'scenes/urls'
 
@@ -189,6 +190,7 @@ export function RenderDataWarehouseSourceIcon({
         azure: Iconazure,
         Salesforce: IconSalesforce,
         MSSQL: IconMSSQL,
+        Vitally: IconVitally,
     }[type]
 
     return (
@@ -203,7 +205,7 @@ export function RenderDataWarehouseSourceIcon({
                 }
             >
                 <Link to={getDataWarehouseSourceUrl(type)}>
-                    <img src={icon} alt={type} height={sizePx} width={sizePx} className="rounded" />
+                    <img src={icon} alt={type} height={sizePx} width={sizePx} className="rounded object-contain" />
                 </Link>
             </Tooltip>
         </div>

frontend/src/types.ts

@@ -3861,6 +3861,7 @@ export const externalDataSources = [
     'Zendesk',
     'Snowflake',
     'Salesforce',
+    'Vitally',
 ] as const
 
 export type ExternalDataSourceType = (typeof externalDataSources)[number]

latest_migrations.manifest

@@ -5,7 +5,7 @@ contenttypes: 0002_remove_content_type_name
 ee: 0016_rolemembership_organization_member
 otp_static: 0002_throttling
 otp_totp: 0002_auto_20190420_0723
-posthog: 0465_datawarehouse_stripe_account
+posthog: 0466_alter_externaldatasource_source_type
 sessions: 0001_initial
 social_django: 0010_uid_db_index
 two_factor: 0007_auto_20201201_1019

mypy-baseline.txt

@@ -42,6 +42,15 @@ posthog/temporal/data_imports/pipelines/rest_source/__init__.py:0: error: Argume
 posthog/temporal/data_imports/pipelines/rest_source/__init__.py:0: error: Incompatible default for argument "resolved_param" (default has type "ResolvedParam | None", argument has type "ResolvedParam")  [assignment]
 posthog/temporal/data_imports/pipelines/rest_source/__init__.py:0: error: Unused "type: ignore" comment  [unused-ignore]
 posthog/temporal/data_imports/pipelines/rest_source/__init__.py:0: error: Argument "module" to "SourceInfo" has incompatible type Module | None; expected Module  [arg-type]
+posthog/temporal/data_imports/pipelines/vitally/__init__.py:0: error: Unused "type: ignore" comment  [unused-ignore]
+posthog/temporal/data_imports/pipelines/vitally/__init__.py:0: error: Unused "type: ignore" comment  [unused-ignore]
+posthog/temporal/data_imports/pipelines/vitally/__init__.py:0: error: Unused "type: ignore" comment  [unused-ignore]
+posthog/temporal/data_imports/pipelines/vitally/__init__.py:0: error: Unused "type: ignore" comment  [unused-ignore]
+posthog/temporal/data_imports/pipelines/vitally/__init__.py:0: error: Unused "type: ignore" comment  [unused-ignore]
+posthog/temporal/data_imports/pipelines/vitally/__init__.py:0: error: Unused "type: ignore" comment  [unused-ignore]
+posthog/temporal/data_imports/pipelines/vitally/__init__.py:0: error: Unused "type: ignore" comment  [unused-ignore]
+posthog/temporal/data_imports/pipelines/vitally/__init__.py:0: error: Unused "type: ignore" comment  [unused-ignore]
+posthog/temporal/data_imports/pipelines/vitally/__init__.py:0: error: Unused "type: ignore" comment  [unused-ignore]
 posthog/utils.py:0: error: No overload variant of "asdict" matches argument type "type[DataclassInstance]"  [call-overload]
 posthog/utils.py:0: note: Possible overload variants:
 posthog/utils.py:0: note: def asdict(obj: DataclassInstance) -> dict[str, Any]

posthog/migrations/0466_alter_externaldatasource_source_type.py

@@ -0,0 +1,30 @@
+# Generated by Django 4.2.15 on 2024-09-05 10:44
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("posthog", "0465_datawarehouse_stripe_account"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="externaldatasource",
+            name="source_type",
+            field=models.CharField(
+                choices=[
+                    ("Stripe", "Stripe"),
+                    ("Hubspot", "Hubspot"),
+                    ("Postgres", "Postgres"),
+                    ("Zendesk", "Zendesk"),
+                    ("Snowflake", "Snowflake"),
+                    ("Salesforce", "Salesforce"),
+                    ("MySQL", "MySQL"),
+                    ("MSSQL", "MSSQL"),
+                    ("Vitally", "Vitally"),
+                ],
+                max_length=128,
+            ),
+        ),
+    ]

posthog/temporal/data_imports/pipelines/schemas.py

@@ -17,6 +17,11 @@
     INCREMENTAL_ENDPOINTS as SALESFORCE_INCREMENTAL_ENDPOINTS,
     INCREMENTAL_FIELDS as SALESFORCE_INCREMENTAL_FIELDS,
 )
+from posthog.temporal.data_imports.pipelines.vitally.settings import (
+    ENDPOINTS as VITALLY_ENDPOINTS,
+    INCREMENTAL_ENDPOINTS as VITALLY_INCREMENTAL_ENDPOINTS,
+    INCREMENTAL_FIELDS as VITALLY_INCREMENTAL_FIELDS,
+)
 
 PIPELINE_TYPE_SCHEMA_DEFAULT_MAPPING = {
     ExternalDataSource.Type.STRIPE: STRIPE_ENDPOINTS,
@@ -29,6 +34,7 @@
     ExternalDataSource.Type.SALESFORCE: SALESFORCE_ENDPOINTS,
     ExternalDataSource.Type.MYSQL: (),
     ExternalDataSource.Type.MSSQL: (),
+    ExternalDataSource.Type.VITALLY: VITALLY_ENDPOINTS,
 }
 
 PIPELINE_TYPE_INCREMENTAL_ENDPOINTS_MAPPING = {
@@ -40,6 +46,7 @@
     ExternalDataSource.Type.SALESFORCE: SALESFORCE_INCREMENTAL_ENDPOINTS,
     ExternalDataSource.Type.MYSQL: (),
     ExternalDataSource.Type.MSSQL: (),
+    ExternalDataSource.Type.VITALLY: VITALLY_INCREMENTAL_ENDPOINTS,
 }
 
 PIPELINE_TYPE_INCREMENTAL_FIELDS_MAPPING: dict[ExternalDataSource.Type, dict[str, list[IncrementalField]]] = {
@@ -51,4 +58,5 @@
     ExternalDataSource.Type.SALESFORCE: SALESFORCE_INCREMENTAL_FIELDS,
     ExternalDataSource.Type.MYSQL: {},
     ExternalDataSource.Type.MSSQL: {},
+    ExternalDataSource.Type.VITALLY: VITALLY_INCREMENTAL_FIELDS,
 }