sqlalchemy 输出到排序字典

Question

我的表如下所示：

User

| id | name  | 

| 1  | user1 |

| 2  | user2 |

| 3  | user3 |

| 4  | user4 |

| 5  | user5 |

| 6  | user6 |

| 7  | user7 |

| 8  | user8 |

Event

| id | name   | created_by | published_by |

| 1  | event1 |     1      |       2      |

| 2  | event2 |     2      |       2      |

| 3  | event3 |     3      |       1      |

Team

| id | name  | player1 | player2 | player3 | player4 |

| 1  | teamA |    1    |    2    |         |         |

| 2  | teamB |    5    |    6    |         |         |

Schedule

| id | event_id | player1 | player2 | team1 | team2 | when_do_they_play   | round |

| 1  |     1    |    1    |    3    |       |       | 2019-09-20 11:22:33 |   1   |

| 1  |     1    |    2    |    4    |       |       | 2019-09-21 12:32:23 |   1   |

| 1  |     1    |    5    |    6    |       |       | 2019-09-22 22:42:03 |   4   |

| 2  |     2    |         |         |   1   |   2   | 2019-09-25 21:12:43 |   2   |

我正在使用以下查询从中选择数据

scheduled_games = (

    db.session.query(

        e.name.label('event_name'),

        u1.name.label('player1'),

        c1.code.label('player1_country'),

        u2.name.label('player2'),

        c2.code.label('player2_country'),

        s.when_do_they_play,

        s.round.label('round'),

        t1.name.label('team1'),

        t2.name.label('team2')

    )

    .outerjoin(u1, u1.id == s.player1)

    .outerjoin(u2, u2.id == s.player2)

    .outerjoin(e, e.id == s.event_id)

    .outerjoin(c1, c1.id == u1.country)

    .outerjoin(c2, c2.id == u2.country)

    .outerjoin(t1, t1.id == s.team1)

    .outerjoin(t2, t2.id == s.team2)

    .filter(s.when_do_they_play>=(datetime.utcnow() - timedelta(days=10)))\

    .order_by(s.when_do_they_play)

    .order_by(s.round.asc())

)

Answer 1

您希望每个事件的回合由列表表示。在您的尝试中，这些行：if ldata[1] or ldata[3]:    scheduled_json[ldata[0]]['round'+str(ldata[6])]['player1']=ldata[1]    scheduled_json[ldata[0]]['round'+str(ldata[6])]['player2']=ldata[3]else:    scheduled_json[ldata[0]]['round'+str(ldata[6])]['team1']=ldata[7]    scheduled_json[ldata[0]]['round'+str(ldata[6])]['team2']=ldata[8]... 将覆盖嵌套在密钥中的任何现有密钥'round' + str(ldata[6])。而不是这个：scheduled_json[ldata[0]]['round'+str(ldata[6])]={}使用list：scheduled_json[ldata[0]]['round'+str(ldata[6])] = []然后为每个事件的每一轮附加一个新dict的：listif ldata[1] or ldata[3]:    d = {'player1': ldata[1], 'player2': ldata[2]}else:    d = {'team1': ldata[7], 'team2': ldata[8]}d['when_do_they_play'] = ldata[5]scheduled_json[ldata[0]]['round'+str(ldata[6])].append(d)从风格上讲，您也可以对整体可读性进行一些改进。例如，您的解决方案可以这样重写：for ldata in scheduled_games:    event = ldata.event  # the row proxy facilitates attribute access syntax    if event not in scheduled_json:        scheduled_json[event] = {}    round_key = f"round{ldata.round}"  # have a look at f-string and str.format() for string concat    if round_key not in scheduled_json[event]:        print('init!')        scheduled_json[event][round_key] = []    if ldata.player1 or ldata.player2:        d = {'player1': ldata.player1, 'player2': ldata.player2}    else:        d = {'team1': ldata.team1, 'team2': ldata.team2}    d['when_do_they_play'] = ldata.when_do_they_play    scheduled_json[event][round_key].append(d)您甚至可以更进一步并使用 acollections.defaultdict这样您就不需要实例化空集合：from collections import defaultdictscheduled_json = defaultdict(lambda: defaultdict(list))for ldata in scheduled_games:    if ldata.player1 or ldata.player2:        d = {'player1': ldata.player1, 'player2': ldata.player2}    else:        d = {'team1': ldata.team1, 'team2': ldata.team2}    d['when_do_they_play'] = ldata.when_do_they_play    scheduled_json[ldata.event][f"round{ldata.round}"].append(d)