Let's see how we can store python objects, this include strings, list, function, etc.., inside a DB like pysonDB
Consider I've function foo
which takes an argument x
and return x * (x + x)
. Which I want to store inside a DB.
def foo(x: float) -> float: return x * (x + x)
First things first we nee to convert this object to byte string. To do that we can use the pickle
. module.
import pickle byte_string = pickle.dumps(foo)
This byte string cannot be directly added to a PysonDB database, as byte strings are not JSON serializable. So a simple workaround will be to add quotes around the byte string which can be easily done like this.
obj_string = f"{byte_string}"
This string can be now added to the DB
from pysondb import db a = db.getDb("test.json") a.add({ "name": "foo_function", "obj": obj_string })
To get the object back we can do the following steps.
data = a.getBy({"name": "foo_function"})
Now data
will look something like this.
[{'name': 'foo_function', 'obj': "b'\\x80\\x04\\x95\\x14\\x00\\x00\\x00\\x00\\x00\\x00\\x00\\x8c\\x08__main__\\x94\\x8c\\x03foo\\x94\\x93\\x94.'", 'id': 316182400052721056}]
To convert the obj string back to a callable function. We can do the following.
str_obj = data[0]["obj"]
Remember that str_obj
is still a string and not a byte string, which is what we need, since the required byte string is inside this string, we can simply evaluate the string.
import ast byte_obj = ast.literal_eval(str_obj)
To call the function we can do this.
call_obj = pickle.loads(byte_obj) print(call_obj(3)) # output 18
Entire Code
import ast import pickle from pysondb import db def foo(x: float) -> float: return x * (x + x) byte_string = pickle.dumps(foo) obj_string = f"{byte_string}" a = db.getDb("test2.json") a.add({ "name": "foo_function", "obj": obj_string }) data = a.getBy({"name": "foo_function"}) str_obj = data[0]["obj"] byte_obj = ast.literal_eval(str_obj) call_obj = pickle.loads(byte_obj) print(call_obj(3))
So we have successfully stored a python object inside a DB. The steps are the same for all the objects like list or dict.
Top comments (0)