Skip to content

Commit

Permalink
Merge branch 'bulk-update-json' of github.com:rabbull/aiida-core into…
Browse files Browse the repository at this point in the history
… bulk-update-json
  • Loading branch information
rabbull committed Dec 10, 2024
2 parents e3a7d0c + e85405f commit c92db47
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 1 deletion.
2 changes: 1 addition & 1 deletion src/aiida/storage/psql_dos/backend.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,11 +10,11 @@

import functools
import gc
import json
import pathlib
from collections import defaultdict
from contextlib import contextmanager, nullcontext
from typing import TYPE_CHECKING, Iterator, Optional, Sequence, Set, Union
import json

from disk_objectstore import Container, backup_utils
from pydantic import BaseModel, Field
Expand Down
3 changes: 3 additions & 0 deletions src/aiida/storage/psql_dos/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@

import json
from typing import TypedDict

from sqlalchemy import event


Expand Down Expand Up @@ -64,9 +65,11 @@ class PsqlConfig(TypedDict, total=False):
$$;
""".strip()


def register_jsonb_patch_function(conn, *args, **kwargs):
print('reg', conn.execute(JSONB_PATCH_FUNCTION))


def create_sqlalchemy_engine(config: PsqlConfig):
"""Create SQLAlchemy engine (to be used for QueryBuilder queries)
Expand Down

0 comments on commit c92db47

Please sign in to comment.