Skip to content

Commit 7329c6e

Browse files
akxc00kiemon5ter
authored andcommitted
Clean up remaining legacy try: except ImportError: blocks
1 parent b185996 commit 7329c6e

9 files changed

+11
-91
lines changed

src/saml2/__init__.py

+3-17
Original file line numberDiff line numberDiff line change
@@ -17,28 +17,14 @@
1717
"""
1818

1919
import logging
20+
from xml.etree import ElementTree
21+
22+
import defusedxml.ElementTree
2023

2124
from saml2.validate import valid_instance
2225
from saml2.version import version as __version__
2326

2427

25-
try:
26-
from xml.etree import ElementTree as ElementTree
27-
28-
if ElementTree.VERSION < "1.3.0":
29-
# cElementTree has no support for register_namespace
30-
# neither _namespace_map, thus we sacrify performance
31-
# for correctness
32-
from xml.etree import ElementTree
33-
except ImportError:
34-
try:
35-
import cElementTree as ElementTree
36-
except ImportError:
37-
from elementtree import ElementTree
38-
39-
import defusedxml.ElementTree
40-
41-
4228
logger = logging.getLogger(__name__)
4329

4430
NAMESPACE = "urn:oasis:names:tc:SAML:2.0:assertion"

src/saml2/pack.py

+3-17
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,9 @@
1818
import logging
1919
from urllib.parse import urlencode
2020
from urllib.parse import urlparse
21+
from xml.etree import ElementTree as ElementTree
22+
23+
import defusedxml.ElementTree
2124

2225
import saml2
2326
from saml2.s_utils import deflate_and_base64_encode
@@ -26,23 +29,6 @@
2629
from saml2.xmldsig import SIG_ALLOWED_ALG
2730

2831

29-
try:
30-
from xml.etree import ElementTree as ElementTree
31-
32-
if ElementTree.VERSION < "1.3.0":
33-
# cElementTree has no support for register_namespace
34-
# neither _namespace_map, thus we sacrify performance
35-
# for correctness
36-
from xml.etree import ElementTree
37-
except ImportError:
38-
try:
39-
import cElementTree as ElementTree
40-
except ImportError:
41-
from elementtree import ElementTree
42-
43-
import defusedxml.ElementTree
44-
45-
4632
logger = logging.getLogger(__name__)
4733

4834
NAMESPACE = "http://schemas.xmlsoap.org/soap/envelope/"

src/saml2/soap.py

+3-12
Original file line numberDiff line numberDiff line change
@@ -6,24 +6,15 @@
66
"""
77
import logging
88
import re
9+
from xml.etree import ElementTree as ElementTree
10+
11+
import defusedxml.ElementTree
912

1013
from saml2 import create_class_from_element_tree
1114
from saml2.samlp import NAMESPACE as SAMLP_NAMESPACE
1215
from saml2.schema import soapenv
1316

1417

15-
try:
16-
from xml.etree import ElementTree as ElementTree
17-
except ImportError:
18-
try:
19-
import cElementTree as ElementTree
20-
except ImportError:
21-
# noinspection PyUnresolvedReferences
22-
from elementtree import ElementTree
23-
24-
import defusedxml.ElementTree
25-
26-
2718
logger = logging.getLogger(__name__)
2819

2920

tests/test_00_xmldsig.py

-6
Original file line numberDiff line numberDiff line change
@@ -7,12 +7,6 @@
77

88
import unittest
99

10-
11-
try:
12-
pass
13-
except ImportError:
14-
pass
15-
1610
import ds_data
1711

1812
import saml2.xmldsig as ds

tests/test_02_saml.py

-4
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,6 @@
55

66
__author__ = "[email protected] (Roland Hedberg)"
77

8-
try:
9-
pass
10-
except ImportError:
11-
pass
128

139
import ds_data
1410
from pytest import raises

tests/test_03_saml2.py

+1-11
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
#!/usr/bin/env python
22

3+
from defusedxml.common import EntitiesForbidden
34
from pytest import raises
45
import saml2_data
56

@@ -16,17 +17,6 @@
1617
from saml2.saml import SubjectLocality
1718

1819

19-
try:
20-
pass
21-
except ImportError:
22-
try:
23-
pass
24-
except ImportError:
25-
pass
26-
27-
from defusedxml.common import EntitiesForbidden
28-
29-
3020
ITEMS = {
3121
NameID: [
3222
"""<?xml version="1.0" encoding="utf-8"?>

tests/test_04_samlp.py

-5
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,6 @@
55

66
__author__ = "[email protected] (Roland Hedberg)"
77

8-
try:
9-
pass
10-
except ImportError:
11-
pass
12-
138
import samlp_data
149

1510
import saml2

tests/test_05_md.py

-6
Original file line numberDiff line numberDiff line change
@@ -5,12 +5,6 @@
55

66
__author__ = "[email protected] (Roland Hedberg)"
77

8-
9-
try:
10-
pass
11-
except ImportError:
12-
pass
13-
148
import ds_data
159
import md_data
1610

tests/test_43_soap.py

+1-13
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,6 @@
11
#!/usr/bin/env python
22

3-
try:
4-
from xml.etree import ElementTree as ElementTree
5-
6-
if ElementTree.VERSION < "1.3.0":
7-
# cElementTree has no support for register_namespace
8-
# neither _namespace_map, thus we sacrify performance
9-
# for correctness
10-
from xml.etree import ElementTree
11-
except ImportError:
12-
try:
13-
import cElementTree as ElementTree
14-
except ImportError:
15-
from elementtree import ElementTree
3+
from xml.etree import ElementTree as ElementTree
164

175
from defusedxml.common import EntitiesForbidden
186
from pytest import raises

0 commit comments

Comments
 (0)