dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-01 02:05:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

feat: Add PostHog analytics integration

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-13 13:03:51 -07:00
parent d7a29c42b7
commit 46489f1a46
1 changed files with 16 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

20
aider/analytics.py
View file

@ -6,15 +6,18 @@ import uuid
from pathlib import Path from pathlib import Path
from mixpanel import Mixpanel from mixpanel import Mixpanel
from posthog import Posthog
from aider import __version__ from aider import __version__
from aider.dump import dump # noqa: F401 from aider.dump import dump # noqa: F401
project_token = "6da9a43058a5d1b9f3353153921fb04d" mixpanel_project_token = "6da9a43058a5d1b9f3353153921fb04d"
posthog_project_api_key = 'phc_99T7muzafUMMZX15H8XePbMSreEUzahHbtWjy3l5Qbv'
posthog_host = 'https://us.i.posthog.com'
class Analytics: class Analytics:
mp = None mp = None
ph = None
user_id = None user_id = None
disable = None disable = None
logfile = None logfile = None
@ -24,6 +27,7 @@ class Analytics:
self.disable = disable self.disable = disable
if not track or disable: if not track or disable:
self.mp = None self.mp = None
self.ph = None
if disable: if disable:
self.mark_as_disabled() self.mark_as_disabled()
return return
@ -31,7 +35,8 @@ class Analytics:
self.user_id = self.get_or_create_uuid() self.user_id = self.get_or_create_uuid()
if self.user_id and not self.disable: if self.user_id and not self.disable:
self.mp = Mixpanel(project_token) self.mp = Mixpanel(mixpanel_project_token)
self.ph = Posthog(project_api_key=posthog_project_api_key, host=posthog_host)
def get_data_file_path(self): def get_data_file_path(self):
data_file = Path.home() / ".aider" / "analytics.json" data_file = Path.home() / ".aider" / "analytics.json"
@ -71,7 +76,7 @@ class Analytics:
} }
def event(self, event_name, main_model=None, **kwargs): def event(self, event_name, main_model=None, **kwargs):
if not self.mp and not self.logfile: if not (self.mp or self.ph) and not self.logfile:
return return
properties = {} properties = {}
@ -97,6 +102,9 @@ class Analytics:
if self.mp: if self.mp:
self.mp.track(self.user_id, event_name, properties) self.mp.track(self.user_id, event_name, properties)
if self.ph:
self.ph.capture(self.user_id, event_name, properties)
if self.logfile: if self.logfile:
log_entry = { log_entry = {
"event": event_name, "event": event_name,
@ -107,3 +115,7 @@ class Analytics:
with open(self.logfile, "a") as f: with open(self.logfile, "a") as f:
json.dump(log_entry, f) json.dump(log_entry, f)
f.write("\n") f.write("\n")
def __del__(self):
if self.ph:
self.ph.shutdown()