Module isort.comments

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

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

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