Is there a way to refactor and move a variable to a new file and update imports?

Ex. I want to move foo to a new file a2.py and have the import in b.py auto update to use that new path.
Is this possible?

```
# a.py

def foo():
    pass
```

```
# b.py

from a import foo

def bar():
    return foo()
```