Skip to content
Open
Show file tree
Hide file tree
Changes from 30 commits
Commits
Show all changes
58 commits
Select commit Hold shift + click to select a range
4d6488c
Initial implementaitons of candidate vs rewrite shuttle
knassre-bodo Oct 9, 2025
5369379
Initial implementation of predicate server integration working on cry…
knassre-bodo Oct 9, 2025
36cab6e
WIP adding to lookup table
knassre-bodo Oct 9, 2025
ed6650c
Rewriting the rest of the filter count queries
knassre-bodo Oct 9, 2025
cc2bbed
Moving server address into mask server info setup
knassre-bodo Oct 9, 2025
a6d4b29
[RUN ALL]
knassre-bodo Oct 9, 2025
beadb15
Adding more tests
knassre-bodo Oct 10, 2025
1b4bcac
Merge branch 'main' into kian/mask_server_rewrite
knassre-bodo Oct 14, 2025
5ea82f1
Switching up relational shuttle handling for simplification
knassre-bodo Oct 15, 2025
f0f512c
Minor adjustments to file placement
knassre-bodo Oct 15, 2025
54ecef1
Moved some logic from rewrite shuttle to candidate visitor
knassre-bodo Oct 15, 2025
557aaeb
Added more tests
knassre-bodo Oct 15, 2025
6b109d9
Added rewrite shuttle docstrings/comments
knassre-bodo Oct 16, 2025
1377916
Adding remaining documentation
knassre-bodo Oct 16, 2025
891c472
Removing dead rule
knassre-bodo Oct 16, 2025
7d7580b
Merge branch 'main' into kian/mask_server_rewrite
knassre-bodo Oct 16, 2025
62db4bf
[RUN ALL]
knassre-bodo Oct 16, 2025
c9f6a59
[RUN ALL]
knassre-bodo Oct 16, 2025
7c37110
Adding logging to keep track of the batch requests sent
knassre-bodo Oct 26, 2025
127244f
Ensuring non-predicate sub-expressions are not sent to the server [RU…
knassre-bodo Oct 26, 2025
1f2dc6d
Ensuring non-predicate sub-expressions are not sent to the server [RU…
knassre-bodo Oct 26, 2025
2864e4a
Merge branch 'main' into kian/mask_server_rewrite
knassre-bodo Oct 26, 2025
b278f9b
Adding date/datetime/timestamp literal handling tests [RUN CI]
knassre-bodo Oct 26, 2025
74b8824
Initial implementation added, as well as early testing
knassre-bodo Oct 27, 2025
7de68bd
Added bubbleprop tests and more warning log tests [RUN CI]
knassre-bodo Oct 28, 2025
d4d2b29
Added docstrings/comments
knassre-bodo Oct 28, 2025
dcbb69c
Added new operators support, need to add new tests for datetime, quar…
knassre-bodo Oct 30, 2025
feabd8a
Added more tests, handled predicate pushdown bug with least/greatest,…
knassre-bodo Oct 30, 2025
940dd16
Added remaining tests [RUN CI]
knassre-bodo Oct 31, 2025
10f40da
Merge branch 'kian/mask_server_rewrite' into kian/unmask_warning_logs
knassre-bodo Oct 31, 2025
a6f6a37
Predicate server revisions with new API
knassre-bodo Nov 5, 2025
af10c5b
JSON request/response reformatting WIP
knassre-bodo Nov 16, 2025
0371ec5
Adding four-phase algorithm, need to implement step #3
knassre-bodo Nov 19, 2025
3996ced
Updating rewrite handling, need to add DP algorithm
knassre-bodo Nov 19, 2025
29e0e3f
Finishing implementation of min cover set
knassre-bodo Nov 21, 2025
f9c05b2
Added edge case tests for selection algorithm
knassre-bodo Nov 21, 2025
4f274fd
Minor test adjustment
knassre-bodo Nov 21, 2025
18379ef
Minor test adjustment
knassre-bodo Nov 21, 2025
f512f8b
Merge branch 'main' into kian/mask_server_rewrite
knassre-bodo Nov 24, 2025
90f0671
Resolving conflicts [RUN ALL]
knassre-bodo Nov 24, 2025
f6a571b
Merge branch 'main' into kian/mask_server_rewrite
knassre-bodo Nov 26, 2025
b728348
Added the FQN slash handling
knassre-bodo Nov 26, 2025
8e03b04
Revisions, QUOTE operator handling, docstrings/documentation [RUN ALL]
knassre-bodo Dec 2, 2025
a3c79cf
Fixing mask server tests [RUN ALL]
knassre-bodo Dec 3, 2025
32d7ee2
API-based revisions overhaul WIP
knassre-bodo Dec 10, 2025
0ed7303
Mask server working, need to iron out kinks with 'retail_transactions…
knassre-bodo Dec 18, 2025
7e98a09
More documentation
knassre-bodo Dec 22, 2025
28c7478
[RUN CI]
knassre-bodo Dec 22, 2025
af7089e
More tests after TS fixed, still need to iterate and remove prints [R…
knassre-bodo Dec 23, 2025
f619356
Edge case debugging WIP
knassre-bodo Dec 23, 2025
8a5f82d
Adding PYDOUGH_MASK_SERVER_PATH to CI
knassre-bodo Dec 23, 2025
aaf9af1
Resolving conflicts
knassre-bodo Dec 23, 2025
4e0cf6f
Added more tests after mask server debugging, identified more bugs in…
knassre-bodo Dec 29, 2025
3cdf093
Bugfixes and breaking up batches to be max length of 16
knassre-bodo Jan 6, 2026
a01a4a8
Merge branch 'main' into kian/mask_server_rewrite
knassre-bodo Jan 6, 2026
d821d1a
[RUN ALL]
knassre-bodo Jan 6, 2026
c0c77e0
Testing fixes [RUN CI][RUN SF_MASKED]
knassre-bodo Jan 6, 2026
962a164
Resolving conflicts
knassre-bodo Jan 7, 2026
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 22 additions & 0 deletions pydough/configs/session.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@
load_database_context,
)
from pydough.errors import PyDoughErrorBuilder
from pydough.mask_server import MaskServerInfo
from pydough.metadata import GraphMetadata, parse_json_metadata_from_file

from .pydough_configs import PyDoughConfigs
Expand All @@ -50,6 +51,7 @@ def __init__(self) -> None:
connection=empty_connection, dialect=DatabaseDialect.ANSI
)
self._error_builder: PyDoughErrorBuilder = PyDoughErrorBuilder()
self._mask_server: MaskServerInfo | None = None

@property
def metadata(self) -> GraphMetadata | None:
Expand Down Expand Up @@ -131,6 +133,26 @@ def error_builder(self, builder: PyDoughErrorBuilder) -> None:
"""
self._error_builder = builder

@property
def mask_server(self) -> MaskServerInfo | None:
"""
Get the active mask server information.

Returns:
The active mask server information.
"""
return self._mask_server

@mask_server.setter
def mask_server(self, server_info: MaskServerInfo | None) -> None:
"""
Set the active mask server information.

Args:
The mask server information to set.
"""
self._mask_server = server_info

def connect_database(self, database_name: str, **kwargs) -> DatabaseContext:
"""
Create a new DatabaseContext and register it in the session. This returns
Expand Down
Loading