sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2025-01-11 05:18:09 +03:00
Code

netcmd: models: EnumField now also supports IntFlag

Browse Source 
Signed-off-by: Rob van der Linde <rob@catalyst.net.nz>
Reviewed-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
This commit is contained in:
Rob van der Linde 2024-02-01 16:52:17 +13:00 committed by Andrew Bartlett
parent 3c8d449ad3
commit 4595a1dae3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/samba/netcmd/domain/models/fields.py
View File

@ -20,7 +20,7 @@
# along with this program. If not, see <http://www.gnu.org/licenses/>. # along with this program. If not, see <http://www.gnu.org/licenses/>.
# #
from enum import IntEnum from enum import IntEnum, IntFlag
import io import io
from abc import ABCMeta, abstractmethod from abc import ABCMeta, abstractmethod
@ -181,7 +181,7 @@ class EnumField(Field):
Has a special case for IntEnum as the constructor only accepts int. Has a special case for IntEnum as the constructor only accepts int.
""" """
if issubclass(self.enum, IntEnum): if issubclass(self.enum, (IntEnum, IntFlag)):
return self.enum(int(str(value))) return self.enum(int(str(value)))
else: else:
return self.enum(str(value)) return self.enum(str(value))