Merge pull request 'feature/remove-statics-from-parser-fuzzer' (#13) from feature/remove-statics-from-parser-fuzzer into master

Reviewed-on: https://git.epenguin.net/gentoo-utils/gentoo-utils-gitea/pulls/13
Reviewed-by: penguin <penguin@epenguin.net>

1 files changed, 2 insertions, 1 deletions

diff --git a/scripts/atom.py b/scripts/atom.py
index 2e8ca7f..a93251a 100755
--- a/scripts/atom.py
+++ b/scripts/atom.py
@@ -2,12 +2,13 @@
 
 import sys
 from portage.dep import Atom
+from portage.exception import InvalidAtom
 
 for line in sys.stdin.buffer:
     try:
         Atom(line.decode().strip())
         sys.stdout.buffer.write(b"0\n")
-    except:
+    except InvalidAtom:
         sys.stdout.buffer.write(b"1\n")
     finally:
         sys.stdout.buffer.flush()