chore: Use warnings.warn(..., category=...) style

open-contracting · May 9, 2024 · fa1f57d · fa1f57d
1 parent 85e7e6b
commit fa1f57d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/ocdskit/combine.py b/ocdskit/combine.py
@@ -88,7 +88,10 @@ def combine_record_packages(packages, uri='', publisher=None, published_date='',
         if 'records' in package:
             output['records'].extend(package['records'])
         else:
-            warnings.warn(f'item {i} has no "records" field (check that it is a record package)')
+            warnings.warn(
+                f'item {i} has no "records" field (check that it is a record package)',
+                category=MissingRecordsWarning,
+            )
         if 'packages' in package:
             output['packages'].update(dict.fromkeys(package['packages']))
 
@@ -122,7 +125,10 @@ def combine_release_packages(packages, uri='', publisher=None, published_date=''
         if 'releases' in package:
             output['releases'].extend(package['releases'])
         else:
-            warnings.warn(f'item {i} has no "releases" field (check that it is a release package)')
+            warnings.warn(
+                f'item {i} has no "releases" field (check that it is a release package)',
+                category=MissingReleasesWarning,
+            )
 
     if publisher:
         output['publisher'] = publisher

diff --git a/ocdskit/packager.py b/ocdskit/packager.py
@@ -195,7 +195,7 @@ def output_records(
                         record['versionedRelease'] = merger.create_versioned_release(releases)
                 except InconsistentTypeError as e:
                     if convert_exceptions_to_warnings:
-                        warnings.warn(MergeErrorWarning(str(e)))
+                        warnings.warn(str(e), category=MergeErrorWarning)
                     else:
                         raise