feature: added dict additional data

Author: n0rfas

git_analytics/engine.py

@@ -1,5 +1,5 @@
 from datetime import date, timezone
-from typing import Dict, Optional
+from typing import Any, Dict, Optional
 
 from git_analytics.entities import AnalyticsResult
 from git_analytics.interfaces import CommitSource
@@ -10,9 +10,11 @@ def __init__(
         self,
         source: CommitSource,
         analyzers_factory,
+        additional_data: Optional[Dict[str, Any]] = None,
     ) -> None:
         self._source = source
         self._analyzers_factory = analyzers_factory
+        self._additional_data = additional_data
 
     def run(
         self,
@@ -36,4 +38,7 @@ def run(
             for analyzer in analyzers:
                 analyzer.process(commit)
 
-        return {analyzer.name: analyzer.result() for analyzer in analyzers}
+        result = {analyzer.name: analyzer.result() for analyzer in analyzers}
+        if self._additional_data:
+            result["additional_data"] = self._additional_data
+        return result

git_analytics/entities.py

@@ -25,7 +25,10 @@ class AnalyticsCommit:
 
 @dataclass
 class AnalyticsResult:
-    def to_dict(self) -> Dict[str, Any]:
+    def __iter__(self):
+        return iter(self._to_dict().items())
+
+    def _to_dict(self) -> Dict[str, Any]:
         return self._make_json_safe(asdict(self))
 
     @staticmethod

git_analytics/main.py

@@ -29,13 +29,15 @@ def make_analyzers():
 def run():
     try:
         repo = Repo()
+        name_branch = repo.active_branch.name
     except InvalidGitRepositoryError:
         print("Error: Current directory is not a git repository.")
         return
 
     engine = CommitAnalyticsEngine(
         source=GitCommitSource(repo),
         analyzers_factory=make_analyzers,
+        additional_data={"name_branch": name_branch},
     )
 
     web_app = create_web_app(engine=engine)

git_analytics/web_app.py

@@ -24,7 +24,7 @@ def on_get_statistics(self, req, resp):
             raise falcon.HTTPBadRequest("Invalid date format", "Use YYYY-MM-DD")
 
         data = self._engine.run(start_date=start_date, stop_date=stop_date)
-        result = {key: value.to_dict() for key, value in data.items()}
+        result = {key: dict(value) for key, value in data.items()}
 
         resp.media = result