Skip to content

Commit e3e593f

Browse files
golorodenclaude
andcommitted
fix: ensure DataFrame has correct columns for empty event streams
When converting an empty event stream, the DataFrame now includes all expected columns instead of being completely empty. This is achieved by explicitly defining the columns parameter when creating the DataFrame. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <[email protected]>
1 parent 8879481 commit e3e593f

File tree

1 file changed

+17
-1
lines changed

1 file changed

+17
-1
lines changed

eventsourcingdb/pandas.py

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,22 @@ async def events_to_dataframe(events: AsyncGenerator[Event, None]) -> pd.DataFra
2020
Returns:
2121
A pandas DataFrame with all event fields as columns
2222
"""
23+
columns = [
24+
"event_id",
25+
"time",
26+
"source",
27+
"subject",
28+
"type",
29+
"data",
30+
"spec_version",
31+
"data_content_type",
32+
"predecessor_hash",
33+
"hash",
34+
"trace_parent",
35+
"trace_state",
36+
"signature",
37+
]
38+
2339
event_list = []
2440

2541
async for event in events:
@@ -40,4 +56,4 @@ async def events_to_dataframe(events: AsyncGenerator[Event, None]) -> pd.DataFra
4056
}
4157
event_list.append(event_dict)
4258

43-
return pd.DataFrame(event_list)
59+
return pd.DataFrame(event_list, columns=columns)

0 commit comments

Comments
 (0)