add to stream object method

Author: raagin

README.md

@@ -1,7 +1,7 @@
 # Django StreamField
 
 This is a simple realisation of StreamField's idea of Wagtail CMS for plain Django admin or with Grappelli skin.
-Stable version: 2.1.0 
+Stable version: 2.1.1
 Django <= 4.2.1
 
 [Major changes (1.4.5 > 2)](changes2.0.md)

frontend/package.json

@@ -1,6 +1,6 @@
 {
   "name": "streamfield_widget.js",
-  "version": "2.1.0",
+  "version": "2.1.1",
   "description": "",
   "private": true,
   "scripts": {

setup.py

@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="django-streamfield",
-    version="2.1.0",
+    version="2.1.1",
     author="Yury Lapshinov",
     author_email="y.raagin@gmail.com",
     description="StreamField for native Django Admin or with Grappelli",
@@ -18,22 +18,17 @@
     python_requires=">=3.7",
     classifiers=[
         'Development Status :: 5 - Production/Stable',
-        'Programming Language :: Python :: 3',
-        'Framework :: Django :: 3.1',
-        'Framework :: Django :: 3.2',
-        'Framework :: Django :: 4.1',
         'License :: OSI Approved :: BSD License',
         'Operating System :: OS Independent',
         'Programming Language :: Python',
         'Programming Language :: Python :: 3',
-        'Programming Language :: Python :: 3.7',
-        'Programming Language :: Python :: 3.8',
         'Programming Language :: Python :: 3.9',
         'Programming Language :: Python :: 3.10',
         "Programming Language :: Python :: 3.11",
         "Framework :: Django",
         "Framework :: Django :: 3.2",
         "Framework :: Django :: 4.0",
         "Framework :: Django :: 4.1",
+        "Framework :: Django :: 4.2",
     ],
 )

streamfield/__init__.py

@@ -1,2 +1,2 @@
 name = "streamfield"
-VERSION = "2.1.0"
+VERSION = "2.1.1"

streamfield/base.py

@@ -137,6 +137,16 @@ def copy(self):
     def to_json(self):
         return json.dumps(self.value)
 
+    def add(self, block):
+        model_class = block.__class__
+        options = _get_default_options(model_class)
+        self.value.append({
+            "id": block.id, 
+            "options": options, 
+            "unique_id": uuid4().hex[:6], 
+            "model_name": model_class.__name__
+        })
+
     @cached_property
     def help_text(self):
         return ADMIN_HELP_TEXT
@@ -197,6 +207,14 @@ def _check_subblocks(obj):
     if changed:
         obj.save()
 
+def _get_default_options(model_class):
+    options = model_class.options if hasattr(model_class, "options") else BLOCK_OPTIONS
+    if hasattr(model_class, "extra_options"):
+        options = deepcopy(options)
+        options.update(model_class.extra_options)
+    options = { k: v['default'] for k, v in options.items() if bool(v.get('default')) }
+    return options
+
 def _copy(model_class, model_str, content, ctx):
     as_list = ctx['as_list']
     resp = dict(
@@ -249,11 +267,7 @@ def migrate_stream_options(stream_obj):
     stream_dict = stream_obj.from_json()
     for b in stream_dict:
         model_class = get_model_by_string(b['model_name'])
-        options = model_class.options if hasattr(model_class, "options") else BLOCK_OPTIONS
-        if hasattr(model_class, "extra_options"):
-            options = deepcopy(options)
-            options.update(model_class.extra_options)
-        options = { k: v['default'] for k, v in options.items() if bool(v.get('default')) }
+        options = _get_default_options(model_class)
         options.update(b['options'])
         b['options'] = options
     return StreamObject(

test_project/requirements.txt

@@ -29,6 +29,21 @@
 ALLOWED_HOSTS = []
 
 
+STREAMFIELD_BLOCK_OPTIONS = {
+    'margin_bottom': {
+        'label': 'Bottom margin',
+        'default': '1',
+        'type': 'select',
+        'options': [
+            {'value': '', 'name': 'No'},
+            {'value': '1', 'name': 'x1'},
+            {'value': '2', 'name': 'x2'},
+            {'value': 'auto', 'name': 'Auto'},
+        ]       
+    }
+}
+
+
 # Application definition
 
 INSTALLED_APPS = [