From 49d92078bd71c056bfa5bb6a26c18e7129ada5f2 Mon Sep 17 00:00:00 2001
From: Shekar Siri <sshekarsiri@gmail.com>
Date: Wed, 31 May 2023 17:31:30 +0200
Subject: [PATCH] feat(ui): support city and state in autocomplete

---
 api/chalicelib/core/autocomplete.py | 2 ++
 api/schemas.py                      | 4 ++++
 2 files changed, 6 insertions(+)

diff --git a/api/chalicelib/core/autocomplete.py b/api/chalicelib/core/autocomplete.py
index 00eff7d61..695c42a25 100644
--- a/api/chalicelib/core/autocomplete.py
+++ b/api/chalicelib/core/autocomplete.py
@@ -16,6 +16,8 @@ def __get_autocomplete_table(value, project_id):
                            schemas.FilterType.user_os,
                            schemas.EventType.custom,
                            schemas.FilterType.user_country,
+                           schemas.FilterType.user_city,
+                           schemas.FilterType.user_state,
                            schemas.EventType.location,
                            schemas.EventType.input]
     autocomplete_events.sort()
diff --git a/api/schemas.py b/api/schemas.py
index d261e1697..e7aacfdb5 100644
--- a/api/schemas.py
+++ b/api/schemas.py
@@ -429,6 +429,8 @@ class FilterType(str, Enum):
     user_browser = "userBrowser"
     user_device = "userDevice"
     user_country = "userCountry"
+    user_city = "userCity"
+    user_state = "userState"
     user_id = "userId"
     user_anonymous_id = "userAnonymousId"
     referrer = "referrer"
@@ -980,6 +982,8 @@ class MetricOfTable(str, Enum):
     user_browser = FilterType.user_browser.value
     user_device = FilterType.user_device.value
     user_country = FilterType.user_country.value
+    user_city = FilterType.user_city.value
+    user_state = FilterType.user_state.value
     user_id = FilterType.user_id.value
     issues = FilterType.issue.value
     visited_url = "location"