From: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
Date: Thu, 1 Apr 2021 02:29:32 +0000 (+1300)
Subject: py/dnsserver add flag from string function
X-Git-Tag: tevent-0.11.0~1245
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=4ce9a0d4879a717fad7830dd70a1f2200d879db2;p=thirdparty%2Fsamba.git

py/dnsserver add flag from string function

Signed-off-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
Reviewed-by: Andreas Schneider <asn@samba.org>
---

diff --git a/python/samba/dnsserver.py b/python/samba/dnsserver.py
index a6e2136b1dc..a9fcb7662e8 100644
--- a/python/samba/dnsserver.py
+++ b/python/samba/dnsserver.py
@@ -287,3 +287,11 @@ def record_from_string(t, data, sep=None, **kwargs):
         raise DNSParseError("Unsupported record type") from e
 
     return Record.from_string(data, sep=sep, **kwargs)
+
+
+def flag_from_string(rec_type):
+    rtype = rec_type.upper()
+    try:
+        return getattr(dnsp, 'DNS_TYPE_' + rtype)
+    except AttributeError:
+        raise DNSParseError('Unknown type of DNS record %s' % rec_type) from e