Move strip_quoted_wrapping function from coder.py to utils.py and update its usage in coder.py.

# Context:
USER: move strip_quoted_wrapping into utils
ASSISTANT: To move the `strip_quoted_wrapping` function from `coder.py` to `utils.py`, you need to make the following changes:

In `coder.py`, remove the `strip_quoted_wrapping` function:

```python
coder.py
<<<<<<< ORIGINAL
    def strip_quoted_wrapping(self, res, fname=None):
        if not res:
            return res

        res = res.splitlines()

        if fname and res[0].strip().endswith(Path(fname).name):
            res = res[1:]

        if res[0].startswith("```") and res[-1].startswith("```"):
            res = res[1:-1]

        res = "\n".join(res)
        if res and res[-1] != "\n":
            res += "\n"

        return res
=======
>>>>>>> UPDATED
```

In `utils.py`, add the `strip_quoted_wrapping` function:

```python
utils.py
<<<<<<< ORIGINAL
=======
def strip_quoted_wrapping(res, fname=None):
    if not res:
        return res

    res = res.splitlines()

    if fname and res[0].strip().endswith(Path(fname).name):
        res = res[1:]

    if res[0].startswith("```") and res[-1].startswith("```"):
        res = res[1:-1]

    res = "\n".join(res)
    if res and res[-1] != "\n":
        res += "\n"

    return res
>>>>>>> UPDATED
```

Finally, update the usage of `strip_quoted_wrapping` in `coder.py` to use the function from `utils.py`:

```python
coder.py
<<<<<<< ORIGINAL
        before_text = self.strip_quoted_wrapping(before_text, fname)
        after_text = self.strip_quoted_wrapping(after_text, fname)
=======
        before_text = utils.strip_quoted_wrapping(before_text, fname)
        after_text = utils.strip_quoted_wrapping(after_text, fname)
>>>>>>> UPDATED
```
This commit is contained in:
Paul Gauthier 2023-05-08 23:54:58 -07:00
parent a232adb43e
commit 05c39b7329

View file

@ -407,9 +407,8 @@ class Coder:
return edited
def do_replace(self, fname, before_text, after_text):
before_text = self.strip_quoted_wrapping(before_text, fname)
after_text = self.strip_quoted_wrapping(after_text, fname)
before_text = utils.strip_quoted_wrapping(before_text, fname)
after_text = utils.strip_quoted_wrapping(after_text, fname)
fname = Path(fname)
# does it want to make a new file?
@ -432,24 +431,6 @@ class Coder:
self.console.print(f"[red]Applied edit to {fname}")
return True
def strip_quoted_wrapping(self, res, fname=None):
if not res:
return res
res = res.splitlines()
if fname and res[0].strip().endswith(Path(fname).name):
res = res[1:]
if res[0].startswith("```") and res[-1].startswith("```"):
res = res[1:-1]
res = "\n".join(res)
if res and res[-1] != "\n":
res += "\n"
return res
def commit(self, history=None, prefix=None, ask=False):
repo = self.repo
if not repo:
@ -611,4 +592,4 @@ def main():
if __name__ == "__main__":
status = main()
sys.exit(status)
sys.exit(status)