summaryrefslogtreecommitdiff
path: root/scripts/dev/ua_fetch.py
blob: 97855411af30ce7109f5325c6903ca0e6936757b (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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
#!/usr/bin/env python3
# vim: ft=python fileencoding=utf-8 sts=4 sw=4 et:

# Copyright 2015-2020 Florian Bruhin (The Compiler) <mail@qutebrowser.org>
# Copyright 2015-2018 lamarpavel
# Copyright 2015-2018 Alexey Nabrodov (Averrin)
#
# This file is part of qutebrowser.
#
# qutebrowser is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# qutebrowser is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with qutebrowser.  If not, see <http://www.gnu.org/licenses/>.


"""Fetch list of popular user-agents.

The script is based on a gist posted by github.com/averrin, the output of this
script is formatted to be pasted into configdata.yml
"""

import requests
from lxml import html  # pylint: disable=import-error


def fetch():
    """Fetch list of popular user-agents.

    Return:
        List of relevant strings.
    """
    url = 'https://techblog.willshouse.com/2012/01/03/most-common-user-agents/'
    page = requests.get(url)
    page = html.fromstring(page.text)
    path = '//*[@id="post-2229"]/div[2]/table/tbody'
    return page.xpath(path)[0]


def filter_list(complete_list, browsers):
    """Filter the received list based on a look up table.

    The LUT should be a dictionary of the format {browser: versions}, where
    'browser' is the name of the browser (eg. "Firefox") as string and
    'versions' is a set of different versions of this browser that should be
    included when found (eg. {"Linux", "MacOSX"}). This function returns a
    dictionary with the same keys as the LUT, but storing lists of tuples
    (user_agent, browser_description) as values.
    """
    # pylint: disable=too-many-nested-blocks
    table = {}
    for entry in complete_list:
        # Tuple of (user_agent, browser_description)
        candidate = (entry[1].text_content(), entry[2].text_content())
        for name in browsers:
            found = False
            if name.lower() in candidate[1].lower():
                for version in browsers[name]:
                    if version.lower() in candidate[1].lower():
                        if table.get(name) is None:
                            table[name] = []
                        table[name].append(candidate)
                        browsers[name].remove(version)
                        found = True
                        break
            if found:
                break
    return table


def main():
    """Generate user agent code."""
    fetched = fetch()
    lut = {
        "Chrome": {"Win10", "Linux"},
    }
    filtered = filter_list(fetched, lut)
    filtered["empty"] = [('', "Use default QtWebKit/QtWebEngine User-Agent")]

    tab = "  "
    for browser in ["Chrome", "empty"]:
        for it in filtered[browser]:
            print('{}- - "{}"'.format(3 * tab, it[0]))
            desc = it[1].replace('\xa0', ' ').replace('  ', ' ')
            print("{}- {}".format(4 * tab, desc))
        print("")


if __name__ == '__main__':
    main()