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