diff --git a/dbml_sqlite/core.py b/dbml_sqlite/core.py index 9dd5a83..f0157d4 100644 --- a/dbml_sqlite/core.py +++ b/dbml_sqlite/core.py @@ -53,10 +53,10 @@ def validDBMLFile(s): Returns: bool: True if s ends with '.dbml', else False. """ - if s.endswith('.dbml'): - return True + if isinstance(s, Path): + return s.suffix == '.dbml' else: - return False + return s.endswith('.dbml') def processFile(target, emulationMode, tableExists=True, indexExists=True, idxNameFunc=uuid.uuid4, join=True): """