aboutsummaryrefslogtreecommitdiff
path: root/xboomx/bin/xboomx_sort.py
blob: 3b78d834d4f6ab7523c18c7a0243fdcd7baf4494 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
#!/usr/bin/env python3
import fileinput

from xboomx.sqlitemgr import get_session, PathItem


def main():
    session = get_session()
    dbitems = session.query(PathItem).all()

    items = {}
    for i in dbitems:
        items[i.name] = i.count

    # read lines and set weight according to db
    items = []
    for input_item in fileinput.input([]):
        input_item = input_item.strip('\n')

        try:
            count = items[input_item]
            items.append((count, input_item))
        except KeyError:
            items.append((0, input_item))

    # sort items
    items.sort(key=lambda x: x[0], reverse=True)

    # print items to be shown on dmenu
    for item in items:
        print(item[1])

    session.close()


if __name__ == '__main__':
    main()