def replace(tokens):
"""
session.query(
...
).filter(
...
).count()
->
session.query(
sqlalchemy.func.count()
).filter(
...
).scalar()
"""
filter_call = tokens["filter call"]
lw, tw = leading_whitespace(filter_call[1:]), trailing_whitespace(
filter_call[:-1])
return ("session.query(" + lw + "sqlalchemy.func.count()" +
("," if "\n" in tw else "") + tw + ").filter" + filter_call +
".scalar()")
def test_leading_whitespace():
assert leading_whitespace(" a ") == " "