refactor: store file in position, rename filename to file_name
Signed-off-by: Konstantin Fickel <mail@konstantinfickel.de>
This commit is contained in:
parent
d5b1541436
commit
79095bad4a
8 changed files with 27 additions and 24 deletions
|
|
@ -37,7 +37,7 @@ def todo() -> None:
|
|||
print(
|
||||
Panel(
|
||||
Markdown(task_shard.markdown_content),
|
||||
title=f"{sharded_document.filename}:{task_shard.start_line}",
|
||||
title=f"{sharded_document.file_name}:{task_shard.start_line}",
|
||||
)
|
||||
)
|
||||
|
||||
|
|
|
|||
|
|
@ -51,12 +51,14 @@ def localize_shard(
|
|||
def localize_stream_file(
|
||||
stream_file: StreamFile, config: RepositoryConfiguration
|
||||
) -> LocalizedShard | None:
|
||||
shard_date = extract_datetime_from_file_name(stream_file.filename)
|
||||
shard_date = extract_datetime_from_file_name(stream_file.file_name)
|
||||
|
||||
if not shard_date or not stream_file.shard:
|
||||
raise ValueError("Could not extract date")
|
||||
|
||||
return localize_shard(stream_file.shard, config, {}, shard_date)
|
||||
return localize_shard(
|
||||
stream_file.shard, config, {"file": stream_file.file_name}, shard_date
|
||||
)
|
||||
|
||||
|
||||
__all__ = ["localize_stream_file"]
|
||||
|
|
|
|||
|
|
@ -26,10 +26,10 @@ def attach_markdown_shard(shard: Shard, markdown_text: str) -> ShardWithMarkdown
|
|||
|
||||
def attach_markdown(file: StreamFile, markdown_text: str) -> StreamFileWithMarkdown:
|
||||
if file.shard is None:
|
||||
return StreamFileWithMarkdown(filename=file.filename, shard=None)
|
||||
return StreamFileWithMarkdown(file_name=file.file_name, shard=None)
|
||||
|
||||
attached_shard = attach_markdown_shard(file.shard, markdown_text)
|
||||
return StreamFileWithMarkdown(filename=file.filename, shard=attached_shard)
|
||||
return StreamFileWithMarkdown(file_name=file.file_name, shard=attached_shard)
|
||||
|
||||
|
||||
__all__ = ["attach_markdown"]
|
||||
|
|
|
|||
|
|
@ -236,7 +236,7 @@ def parse_markdown_file(file_name: str, file_content: str) -> StreamFile:
|
|||
):
|
||||
shard = parsed_shard
|
||||
|
||||
return StreamFile(shard=shard, filename=file_name)
|
||||
return StreamFile(shard=shard, file_name=file_name)
|
||||
|
||||
|
||||
__all__ = ["Shard", "StreamFile", "parse_markdown_file"]
|
||||
|
|
|
|||
|
|
@ -1,4 +1,5 @@
|
|||
from __future__ import annotations
|
||||
|
||||
from pydantic import BaseModel
|
||||
|
||||
|
||||
|
|
@ -11,7 +12,7 @@ class Shard(BaseModel):
|
|||
|
||||
|
||||
class StreamFile(BaseModel):
|
||||
filename: str
|
||||
file_name: str
|
||||
shard: Shard | None = None
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue