Add line number (#18)

* Add line number * Reformat code
home-assistant-ecosystem · Jan 24, 2024 · 7d5fd96 · 7d5fd96
1 parent e3f4b6d
commit 7d5fd96
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -99,3 +99,6 @@ ENV/
 
 # mypy
 .mypy_cache/
+
+# PyCharm
+.idea
diff --git a/CHANGES.txt b/CHANGES.txt
@@ -1,10 +1,11 @@
 Changes
 =======
 
-20231213 - 0.4.1
+20240113 - 0.5.0
 ----------------
 
 - Add KeyError exception (thanks @miaucl)
+- Add line numbers (thanks @polgarc)
 
 20231120 - 0.4.0
 ----------------

diff --git a/opendata_transport/__init__.py b/opendata_transport/__init__.py
@@ -252,7 +252,11 @@ def get_connection(connection):
         connection_info["number"] = ""
         for section in connection["sections"]:
             if section["journey"] is not None:
-                connection_info["number"] = section["journey"]["name"]
+                journey = section["journey"]
+                connection_info["number"] = journey["name"]
+                connection_info["line"] = "".join(
+                    filter(None, [journey["category"], journey["number"]])
+                )
                 break
 
         connection_info["platform"] = connection["from"]["platform"]
-Original file line number
+Diff line change
@@ Expand Up / @@ -99,3 +99,6 @@ ENV/ @@
     # mypy
     .mypy_cache/
+    # PyCharm
+    .idea