mirror of
https://github.com/PostHog/posthog.git
synced 2024-11-24 18:07:17 +01:00
1a6207d791
* Upload static cohort using CSV, closes #2868 * fix tests * Fix tests * Fix e2e test * Avoid double inserts * Speed up query * Move to params * fix tests * Use JSON instead of protobuf * unused import * produce instead of produce_proto * Insert directly into clickhouse, no need for kafka * a few quick fixes * insert in batch to clickhouse * test SQLi since we can't really trust what we are inserting * Extra check for duplicates Co-authored-by: James Greenhill <fuziontech@gmail.com>
28 lines
760 B
Python
28 lines
760 B
Python
from django.core.management.base import BaseCommand
|
|
from infi.clickhouse_orm import Database # type: ignore
|
|
|
|
from posthog.settings import (
|
|
CLICKHOUSE_DATABASE,
|
|
CLICKHOUSE_HTTP_URL,
|
|
CLICKHOUSE_PASSWORD,
|
|
CLICKHOUSE_USERNAME,
|
|
CLICKHOUSE_VERIFY,
|
|
)
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Migrate clickhouse"
|
|
|
|
def handle(self, *args, **options):
|
|
try:
|
|
Database(
|
|
CLICKHOUSE_DATABASE,
|
|
db_url=CLICKHOUSE_HTTP_URL,
|
|
username=CLICKHOUSE_USERNAME,
|
|
password=CLICKHOUSE_PASSWORD,
|
|
verify_ssl_cert=False,
|
|
).migrate("ee.clickhouse.migrations")
|
|
print("migration successful")
|
|
except Exception as e:
|
|
print(e)
|