mirror of
https://github.com/PiBrewing/craftbeerpi4.git
synced 2024-12-26 23:41:47 +01:00
32 lines
933 B
Python
32 lines
933 B
Python
from typing import List, Optional, Tuple
|
|
|
|
|
|
def parse(line: str) -> Tuple[str, str]:
|
|
"""Parses import lines for comments and returns back the
|
|
import statement and the associated comment.
|
|
"""
|
|
comment_start = line.find("#")
|
|
if comment_start != -1:
|
|
return (line[:comment_start], line[comment_start + 1 :].strip())
|
|
|
|
return (line, "")
|
|
|
|
|
|
def add_to_line(
|
|
comments: Optional[List[str]],
|
|
original_string: str = "",
|
|
removed: bool = False,
|
|
comment_prefix: str = "",
|
|
) -> str:
|
|
"""Returns a string with comments added if removed is not set."""
|
|
if removed:
|
|
return parse(original_string)[0]
|
|
|
|
if not comments:
|
|
return original_string
|
|
|
|
unique_comments: List[str] = []
|
|
for comment in comments:
|
|
if comment not in unique_comments:
|
|
unique_comments.append(comment)
|
|
return f"{parse(original_string)[0]}{comment_prefix} {'; '.join(unique_comments)}"
|