Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def test_normal(self, value, replace_text, expected):
assert replace_symbol(value, replace_text) == expected
def test_abnormal(self, value, expected):
with pytest.raises(expected):
replace_symbol(value)
def test_normal_csv(self, tmpdir, file_path, format_name):
filename = pv.replace_symbol(file_path, "")
p_file_path = Path(str(tmpdir.join(filename + Path(file_path).ext)))
p_file_path.parent.makedirs_p()
with open(p_file_path, "w") as f:
f.write(
dedent(
"""\
"attr_a","attr_b","attr_c"
1,4,"a"
2,2.1,"bb"
3,120.9,"ccc"
"""
)
)
expected_list = [
def test_normal_consecutive(
self, value, replace_text, is_replace_consecutive_chars, is_strip, expected
):
assert (
replace_symbol(value, replace_text, is_replace_consecutive_chars, is_strip) == expected
)
def _preprocess_table_name(self):
try:
new_name = pv.sanitize_filename(self._tabledata.table_name, replacement_text="_")
except TypeError:
raise NameValidationError(
"table name must be a string: actual='{}'".format(self._tabledata.table_name)
)
new_name = pv.replace_unprintable_char(new_name, replacement_text="")
new_name = pv.replace_symbol(new_name, replacement_text="_")
new_name = new_name.replace(" ", "_")
new_name = re.sub("_+", "_", new_name)
new_name = new_name.strip("_")
return new_name
from simplesqlite import SQLiteTableDataSanitizer
if dup_col_handler is None:
dup_col_handler = DEFAULT_DUP_COL_HANDLER
normalized_table_data = SQLiteTableDataSanitizer(
table_data, dup_col_handler=dup_col_handler, is_type_inference=self._is_type_inference
).normalize()
if self._symbol_replace_value is None:
return normalized_table_data
return TableData(
normalized_table_data.table_name,
[
replace_symbol(
replace_unprintable_char(header),
self._symbol_replace_value,
is_replace_consecutive_chars=True,
is_strip=True,
)
for header in normalized_table_data.headers
],
normalized_table_data.rows,
dp_extractor=normalized_table_data.dp_extractor,
type_hints=table_data.dp_extractor.column_type_hints,
)
def make_temp_file_path_from_url(temp_dir_path, url):
try:
url_path = urlparse(url).path
except AttributeError:
raise InvalidFilePathError("url must be a string")
if typepy.is_null_string(url_path):
raise InvalidFilePathError("invalid URL path: {}".format(url_path))
temp_name = os.path.basename(url_path.rstrip("/"))
if typepy.is_null_string(temp_name):
temp_name = pathvalidate.replace_symbol(temp_name, replacement_text="_")
if typepy.is_null_string(temp_name):
raise InvalidFilePathError("invalid URL: {}".format(url))
try:
return posixpath.join(temp_dir_path, temp_name)
except (TypeError, AttributeError):
raise InvalidFilePathError("temp_dir_path must be a string")