Merge pull request #120 from k-dal/escape-single-quote-formula-field

Escape single quote formula field
gtalarico · Jul 22, 2021 · bf130cc · bf130cc
2 parents fc0efea + 31efa3b
commit bf130cc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/airtable/params.py b/airtable/params.py
@@ -17,6 +17,7 @@
 """  #
 
 from collections import OrderedDict
+import re
 
 
 class _BaseParam(object):
@@ -205,6 +206,7 @@ def from_name_and_value(field_name, field_value):
             Creates a formula to match cells from from field_name and value
             """
             if isinstance(field_value, str):
+                field_value = re.sub("(?<!\\\\)'", "\\'", field_value)
                 field_value = "'{}'".format(field_value)
 
             formula = "{{{name}}}={value}".format(name=field_name, value=field_value)

diff --git a/tests/test_params.py b/tests/test_params.py
@@ -94,6 +94,12 @@ def test_formula_from_name_and_value():
     formula = AirtableParams.FormulaParam.from_name_and_value("COL", "VAL")
     assert formula == r"{COL}='VAL'"
 
+    formula = AirtableParams.FormulaParam.from_name_and_value("COL", "'VAL'")
+    assert formula == r"{COL}='\'VAL\''"
+
+    formula = AirtableParams.FormulaParam.from_name_and_value("COL", "\\'VAL\\'")
+    assert formula == r"{COL}='\'VAL\''"
+
     formula = AirtableParams.FormulaParam.from_name_and_value("COL", 8)
     assert formula == r"{COL}=8"