| 1234567891011121314151617181920212223242526272829303132 |
- 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)}"
|
