feature: adds handling CE and EE versions

This commit is contained in:
ievgenii vdovenko 2025-06-25 10:52:12 +02:00
parent b5c53fab64
commit edf6be86a1

View file

@ -33,6 +33,8 @@ import collections
import dataclasses as dataclasses
from copy import deepcopy
from pyramid.threadlocal import get_current_registry
from rhodecode.lib.pyramid_utils import get_current_request
from rhodecode.lib.vcs.nodes import FileNode
@ -2227,7 +2229,11 @@ class PullRequestModel(BaseModel):
if repo_type not in ["hg", "git"]:
return False
if pull_request.settings and "close_branch_before_merging" in pull_request.settings:
registry = get_current_registry()
is_enterprise = registry.settings.get("rhodecode.edition_id") == "EE"
if is_enterprise and pull_request.settings and "close_branch_before_merging" in pull_request.settings:
# this feature is available only for the EE edition
return pull_request.settings["close_branch_before_merging"]
key = "rhodecode_{}_close_branch_before_merging".format(repo_type)