diff --git a/pydbml/__init__.py b/pydbml/__init__.py index a585105..868de24 100644 --- a/pydbml/__init__.py +++ b/pydbml/__init__.py @@ -1,3 +1,15 @@ +import os + from . import _classes from .parser import PyDBML from .database import Database + +load = PyDBML.parse_file +loads = PyDBML.parse + +def dump(db: Database, fp: str | os.PathLike): + with open(fp, 'w') as f: + f.write(db.dbml) + +def dumps(db: Database) -> str: + return db.dbml \ No newline at end of file