Freeze your python objects
Project description
gelidum
Freeze your objects in python.
Public domain photo by Katsujiro Maekawa
Latin | English |
---|---|
Caelum est hieme frigidum et gelidum; myrtos oleas quaeque alia assiduo tepore laetantur, aspernatur ac respuit; laurum tamen patitur atque etiam nitidissimam profert, interdum sed non saepius quam sub urbe nostra necat. | In winter the air is cold and frosty: myrtles, olives and all other trees which require constant warmth for them to do well, the climate rejects and spurns, though it allows laurel to grow, and even brings it to a luxuriant leaf. Occasionally, however, it kills it, but that does not happen more frequently than in the neighbourhood of Rome. |
The Letters of the Younger Pliny, First Series — Volume 1 by the Younger Pliny, translated to English by John Benjamin Firth.
Introduction
Inspired by the method freeze found in other languages like Javascript, this package tries to make immutable objects to make it easier avoiding accidental modifications in your code.
See more comments about this project in this Show HN.
Major highlights
- freeze method creates objects with the same attributes of inputs that cannot be expanded or modified.
- Frozen object creation is thread-safe.
- Structural sharing: any frozen object is shared by all of its user objects. There is no copy performed, only reference.
- cpython and pypy support.
How it works
In case of the builtin types (bool, None, int, float, bytes, complex, str) it does nothing, as they are already immutable.
For the list type, a frozenlist with frozen items is returned.
Tuples are already immutable, so a new tuple with frozen items is returned.
When freezing a set, a frozenzet of frozen items is returned.
In the case of dicts, freezing one of them creates a new frozendict with the keys and frozen values of the original dict.
This package, change the methods __setattr__, __delattr__, __set__, __setitem__, and __delitem__ of the object argument and all of its attributed recursively, making them raise an exception if the developer tries to call them to modify the attributes of the instance.
How to use it
Freeze in the same object
from typing import List
from gelidum import freeze
class Dummy(object):
def __init__(self, attr1: int, attr2: List):
self.attr1 = attr1
self.attr2 = attr2
dummy = Dummy(1, [2, 3, 4])
frozen_dummy = freeze(dummy, on_freeze="inplace")
assert(id(dummy) == id(frozen_dummy))
# Both raise exception
new_value = 1
dummy.attr1 = new_value
frozen_dummy.attr1 = new_value
# Both raise exception
new_value_list = [1]
dummy.attr2 = new_value_list
frozen_dummy.attr2 = new_value_list
Freeze in a new object
Basic use
from typing import List
from gelidum import freeze
class Dummy(object):
def __init__(self, attr1: int, attr2: List):
self.attr1 = attr1
self.attr2 = attr2
dummy = Dummy(1, [2, 3, 4])
# on_freeze="copy" by default
frozen_dummy = freeze(dummy)
assert(id(dummy) != id(frozen_dummy))
# on_freeze="copy" by default
frozen_object_dummy2 = freeze(dummy, on_freeze="copy")
# It doesn't raise an exception,
# dummy keeps being a mutable object
new_attr1_value = 99
dummy.attr1 = new_attr1_value
# Raises exception,
# frozen_dummy is an immutable object
frozen_dummy.attr1 = new_attr1_value
Access to original object
Passing flag to freezer
If you are freezing custom objects, you can pass the flag save_original_on_copy
to the freeze method to
ensure you have an attribute original_obj in the frozen method.
from gelidum import freeze
class Dummy(object):
def __init__(self, attr1: int, attr2: int):
self.attr1 = attr1
self.attr2 = attr2
dummy = Dummy(1, 2)
frozen_dummy = freeze(dummy, save_original_on_copy=True)
# We are copying the object and freezing it:
assert(id(dummy) != id(frozen_dummy))
# But we are keeping the original object inside it:
assert(id(dummy) == id(frozen_dummy.original_obj))
Custom freezer
The parameter on_freeze admits a callable, so you can have some side effects when freezing objects.
There is a particular callable class that allows returning the original object:
from gelidum import freeze
from gelidum.on_freeze import OnFreezeOriginalObjTracker
class Dummy(object):
def __init__(self, value1: int, value2: int):
self.attr1 = value1
self.attr2 = value2
dummy = Dummy(value1=1, value2=2)
freezer = OnFreezeOriginalObjTracker()
frozen_dummy = freeze(dummy, on_freeze=freezer)
original_obj = freezer.original_obj
assert(dummy == original_obj)
Note that in the earlier case the original object is not frozen but a copy of it.
What to do when trying to update an attribute
import logging
from gelidum import freeze
class SharedState(object):
def __init__(self, count: int):
self.count = count
shared_state = SharedState(1)
# on_update="exception": raises an exception when an update is tried
frozen_shared_state = freeze(shared_state, on_update="exception")
frozen_shared_state.count = 4 # Raises exception
# on_update="warning": shows a warning in console exception when an update is tried
frozen_shared_state = freeze(shared_state, on_update="warning")
frozen_shared_state.count = 4 # Shows a warning in console
# on_update="nothing": does nothing when an update is tried
frozen_shared_state = freeze(shared_state, on_update="nothing")
frozen_shared_state.count = 4 # Does nothing, as this update did not exist
# on_update=<lambda message, *args, **kwargs>: calls the function
# Note the parameters of that function must be message, *args, **kwargs
frozen_shared_state = freeze(
shared_state,
on_update=lambda message, *args, **kwargs: logging.warning(message)
)
frozen_shared_state.count = 4 # Calls on_update function and logs in the warning level:
# "Can't assign 'count' on immutable instance"
Freeze input params
Use the decorator freeze_params to freeze the input parameters and avoid non-intended modifications:
from typing import List
from gelidum import freeze_params
@freeze_params()
def append_to_list(a_list: List, new_item: int):
a_list.append(new_item)
If freeze_params is called without arguments, all input parameters will be frozen. Otherwise, passing a set of parameters will inform the decorator of which named parameters must be frozen.
from typing import List
from gelidum import freeze_params
@freeze_params(params={"list1", "list2"})
def concat_lists(dest: List, list1: List, list2: List) -> List:
dest = list1 + list2
return dest
# Freeze dest, list1 and list2
concat_lists([], list1=[1, 2, 3], list2=[4, 5, 6])
# Freeze list1 and list2
concat_lists(dest=[], list1=[1, 2, 3], list2=[4, 5, 6])
Always use kwargs unless you want to freeze the args params. A good way to enforce this is by making the function have keyword-only arguments:
from typing import List
from gelidum import freeze_params
@freeze_params(params={"list1", "list2"})
def concat_lists_in(*, dest: List, list1: List, list2: List):
dest = list1 + list2
return dest
You can use the Final typehint from gelidum to signal that an argument is immutable:
from typing import List
from gelidum import freeze_final, Final
@freeze_final
def concatenate_lists(list1: Final[List], list2: Final[List]):
return list1 + list2
Finally, take in account that all freezing is done in a new object (i.e. freeze with on_freeze="copy"). It makes no sense to freeze a parameter of a function that could be used later, outside said function.
Check original (i.e. "hot") class
- get_gelidum_hot_class_name: returns the name of hot class.
- get_gelidum_hot_class_module returns the module reference where the hot class was.
Collections
There are four immutable collections in the gelidum.collections module.
- frozendict
- frozenlist
- frozenzet (frozenset is already a builtin type in Python)
All of these classes can be used to make sure a collection of objects is not modified. Indeed, when creating a new collection object, you can pass a custom freeze function, to customize the freezing process of each of its items, e.g.:
import logging
from gelidum.freeze import freeze
from gelidum.collections import frozenzet
from gelidum.typing import FrozenType
from typing import Any
def my_freeze_func(item: Any) -> FrozenType:
logging.debug(f"Freezing item {item}")
return freeze(item, on_update="exception", on_freeze="copy")
frozen_zet = frozenzet([1, 2, 3], freeze_func=my_freeze_func)
Rationale and background information
Inspired by my old work with Ruby on Rails, I decided to create a mechanism to make objects immutable in Python. The first aim was to do a tool to avoid accidental modifications on the objects while passing them through an execution flow.
Anyways, as time passed I thought that an implementation of a programming language with real threading support (i.e. not cpython) could be benefited from this feature. I know that both cpython and pypy implementations of the Python programming language have a GIL but IronPython and Graalpython don't. IronPython3 has no support for typehintings yet, but Graalpython seems to work fine, so more experiments will be coming.
On the other hand, I'm also interested in creating functional data structures in this package, easing the life of developers that do not want side effects.
It's true that the complexity of Python does not play well with this kind of library. Thus, Python usually serves as easy interface with native libraries (pandas, numpy, etc.) However, this project is fun to develop and maybe with the popularity of alternative implementations of Python some work can be done to improve performance.
More information can be seen in this Show HN post and some appreciated feedback of the users of that great community.
Limitations
- dict, list, tuple and set objects cannot be modified inplace although the flag inplace is set.
- file handler attributes are not supported. An exception is raised when trying to freeze an object with them.
- frozen objects cannot be serialized with marshal.
- frozen objects cannot be (deep)-copied. This limitation is intended to make structural sharing easier.
- Classes with __slots__:
- cannot be frozen in-place.
- will be frozen with a unique class. The frozen class will not be shared by instances of the same class.
Advice & comments on use
On_update parameter of freeze function
Use on_update with a callable to store when somebody tried to write in the immutable object:
import datetime
import logging
import threading
from gelidum import freeze
class Dummy(object):
def __init__(self, attr: int):
self.attr = attr
class FrozenDummyUpdateTryRecorder:
LOCK = threading.Lock()
written_tries = []
@classmethod
def add_writing_try(cls, message, *args, **kwargs):
logging.warning(message)
with cls.LOCK:
cls.written_tries.append({
"message": message,
"args": args,
"kwargs": kwargs,
"datetime": datetime.datetime.utcnow()
})
dummy = Dummy(1)
frozen_dummy = freeze(
dummy,
on_update=FrozenDummyUpdateTryRecorder.add_writing_try
)
# It will call FrozenDummyUpdateTryRecorder.add_writing_try
# and will continue the execution flow with the next sentence.
frozen_dummy.attr = 4
On_freeze parameter of freeze function
The parameter on_freeze of the function freeze must be a string or a callable. This parameter informs of what to do with the object that will be frozen. Should it be the same input object frozen or a copy of it?
If it has a string as parameter, values "inplace" and "copy" are allowed. A value of "inplace" will make the freeze method to try to freeze the object as-is, while a value of "copy" will make a copy of the original object and then, freeze that copy. These are the recommended parameters.
On the other hand, the interesting part is to define a custom on_freeze method. This method must return an object of the same type of the input. This returned will be frozen, and returned to the caller of freeze.
Note this parameter has no interference with the structural sharing of the frozen objects. Any frozen object that have several references to it will be shared, not copied.
import copy
def on_freeze(self, obj: object) -> object:
frozen_object = copy.deepcopy(obj)
# log, copy the original method or do any other
# custom action in this function
return frozen_object
As seen earlier, there is also the possibility to pass a callable object. If you would like you can even define your own on_freeze functions by inheriting from classes:
- OnFreezeCopier
- OnFreezeIdentityFunc
See some examples in on_freeze.py file.
Dependencies
This package has no dependencies.
Roadmap
- Freeze only when attributes are modified? Not exactly but structural sharing is used.
- Include immutable collections.
- Graalpython support.
- Make some use-cases with threading/async module (i.e. server)
Collaborations
This project is open to collaborations. Make a PR or an issue, and I'll take a look to it.
License
MIT license, but if you need any other contact me.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
File details
Details for the file gelidum-0.7.1.tar.gz
.
File metadata
- Download URL: gelidum-0.7.1.tar.gz
- Upload date:
- Size: 32.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.12.7
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 0853e0b7802626b1cf8f3ff8e1d08ee65c9925f4a62d089684f55c9ff1d2b75e |
|
MD5 | 942b4594da1af8c11607f5f1031ed82c |
|
BLAKE2b-256 | 1bf550767eb072cee38c3ce8af0c635c7668499bb51f476163d120cdb8f82a50 |