From 52d59ad2044f963aa34ba6ceca00180595705fa0 Mon Sep 17 00:00:00 2001 From: samhaese <63563702+samhaese@users.noreply.github.com> Date: Fri, 18 Apr 2025 14:46:30 -0700 Subject: [PATCH] Add load(s) and dump(s) methods --- pydbml/__init__.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) 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