Parse raw numbers as steamids

While most of the time we need to parse strings into steamids, sometimes we already have a parsed steamid ready to use. To make handling this case easier, add a try-catch around our regex matching. Signed-off-by: Sean Anderson <[email protected]>
Forty-Bot · Aug 27, 2021 · e588e41 · e588e41
1 parent 997e770
commit e588e41
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/steamid.py b/steamid.py
@@ -1,5 +1,4 @@
-# SPDX-License-Identifier: MIT
-# SPDX-License-Identifier: AGPL-3.0-only
+# SPDX-License-Identifier: MIT AND AGPL-3.0-only
 # Copyright (C) 2020 Sean Anderson <[email protected]>
 # Copyright (c) 2017 Nate the great
 # Copyright (c) 2015 Alexander Corn
@@ -69,8 +68,13 @@ def __init__(self, input):
 
         reg = re.compile("^STEAM_([0-5]):([0-1]):([0-9]+)$")
         reg3 = re.compile("^\[([a-zA-Z]):([0-5]):([0-9]+)(:[0-9]+)?\]")
-        mat = reg.match(input)
-        mat3 = reg3.match(input)
+        try:
+            mat = reg.match(input)
+            mat3 = reg3.match(input)
+        except TypeError:
+            mat = None
+            mat3 = None
+
         if mat:
             self.universe = int(mat[1]) or SteamID.Universe["PUBLIC"]
             self.type = SteamID.Type["INDIVIDUAL"]