diff --git a/dota.py b/dota.py index 73b0b77..ec7a77e 100644 --- a/dota.py +++ b/dota.py @@ -139,11 +139,15 @@ class Friend(BaseModel): } for match_ in matches[:limit]: duration = '%d:%02d:%02d' % utils.convert_seconds_to_hms(match_['duration']) + if match_['party_size'] == None: + if Match.filter(match_id=match_['match_id']).exists(): + match_['party_size'] = Match.get(match_id=match_['match_id']).party_size + else: + match_['party_size'] = 0 + summary = f"{duration}" if match_['party_size'] and match_['party_size'] > 1: summary = f"{match_['party_size']}黑 {duration}" - elif match_['party_size'] == None: - summary = f"??黑 {duration}" - else: + elif match_['party_size'] and match_['party_size'] == 1: summary = f"单排 {duration}" if match_['average_rank']: @@ -228,10 +232,11 @@ def serialize_match_for_discord(match_): if Match.filter(match_id=match_['match_id']).exists(): match_['party_size'] = Match.get(match_id=match_['match_id']).party_size else: + match_['party_size'] = 0 summary = f"{match_['duration']}" if match_['party_size'] > 1: summary = f"{match_['party_size']}黑 {match_['duration']}" - else: + elif match_['party_size'] == 1: summary = f"单排 {match_['duration']}" radiant = []