Reworked rule export

export.py

@@ -32,15 +32,14 @@ if __name__ == '__main__':
 
     DB = database.Database()
 
-    if args.rules:
-        if args.count:
-            print(DB.count_rules(first_party_only=args.first_party))
-        else:
+    if args.count:
+        print(DB.count_records(
+            first_party_only=args.first_party,
+            rules_only=args.rules))
+    else:
+        if args.rules:
             for line in DB.list_rules():
                 print(line)
-    else:
-        if args.count:
-            raise NotImplementedError
         for domain in DB.export(
             first_party_only=args.first_party,
             end_chain_only=args.end_chain,