typing

PostHog · Mar 26, 2024 · 7c9f4a1 · 7c9f4a1
1 parent 3307e5f
commit 7c9f4a1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/posthog/hogql/autocomplete.py b/posthog/hogql/autocomplete.py
@@ -216,9 +216,9 @@ def resolve_table_field_traversers(table: Table, context: HogQLContext) -> Table
         current_table_or_field: FieldOrTable = new_table
         for chain in field.chain:
             if isinstance(current_table_or_field, Table):
-                chain_field = current_table_or_field.fields.get(chain)
+                chain_field = current_table_or_field.fields.get(str(chain))
             elif isinstance(current_table_or_field, LazyJoin):
-                chain_field = current_table_or_field.resolve_table(context).fields.get(chain)
+                chain_field = current_table_or_field.resolve_table(context).fields.get(str(chain))
             elif isinstance(current_table_or_field, DatabaseField):
                 chain_field = current_table_or_field
             else:

diff --git a/posthog/hogql/database/models.py b/posthog/hogql/database/models.py
@@ -75,8 +75,8 @@ class Table(FieldOrTable):
     def has_field(self, name: str | int) -> bool:
         return str(name) in self.fields
 
-    def get_field(self, name: str) -> FieldOrTable:
-        if self.has_field(name):
+    def get_field(self, name: str | int) -> FieldOrTable:
+        if self.has_field(str(name)):
             return self.fields[name]
         raise Exception(f'Field "{name}" not found on table {self.__class__.__name__}')