Skip to content

Commit d0c1094

Browse files
committed
Ensure all results are PureMagicWithConfidence type
1 parent 23c3462 commit d0c1094

File tree

1 file changed

+20
-2
lines changed

1 file changed

+20
-2
lines changed

puremagic/main.py

Lines changed: 20 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -473,7 +473,16 @@ def run_deep_scan(
473473
raise
474474
else:
475475
if result:
476-
return [result]
476+
return [
477+
PureMagicWithConfidence(
478+
confidence=result.confidence,
479+
byte_match=None,
480+
offset=None,
481+
extension=result.extension,
482+
mime_type=result.mime_type,
483+
name=result.name,
484+
)
485+
]
477486
if raise_on_none:
478487
raise PureError("Could not identify file")
479488

@@ -504,7 +513,16 @@ def run_deep_scan(
504513
pass
505514
else:
506515
if result and result.confidence > matches[0].confidence:
507-
return [result]
516+
return [
517+
PureMagicWithConfidence(
518+
confidence=result.confidence,
519+
byte_match=None,
520+
offset=None,
521+
extension=result.extension,
522+
mime_type=result.mime_type,
523+
name=result.name,
524+
)
525+
]
508526
return matches
509527

510528

0 commit comments

Comments
 (0)