psutil

1
#!/usr/bin/env python3
2

3
# Copyright (c) 2009, Giampaolo Rodola'. All rights reserved.
4
# Use of this source code is governed by a BSD-style license that can be
5
# found in the LICENSE file.
6

7
"""Remove raw HTML from README.rst to make it compatible with PyPI on
8
dist upload.
9
"""
10

11
import argparse
12
import re
13

14

15
summary = """\
16
Quick links
17
===========
18

19
- `Home page <https://github.com/giampaolo/psutil>`_
20
- `Install <https://github.com/giampaolo/psutil/blob/master/INSTALL.rst>`_
21
- `Documentation <http://psutil.readthedocs.io>`_
22
- `Download <https://pypi.org/project/psutil/#files>`_
23
- `Forum <http://groups.google.com/group/psutil/topics>`_
24
- `StackOverflow <https://stackoverflow.com/questions/tagged/psutil>`_
25
- `Blog <https://gmpy.dev/tags/psutil>`_
26
- `What's new <https://github.com/giampaolo/psutil/blob/master/HISTORY.rst>`_
27
"""
28

29
funding = """\
30
Sponsors
31
========
32

33
.. image:: https://github.com/giampaolo/psutil/raw/master/docs/_static/tidelift-logo.png
34
  :width: 200
35
  :alt: Alternative text
36

37
`Add your logo <https://github.com/sponsors/giampaolo>`__.
38

39
Example usages"""  # noqa
40

41

42
def main():
43
    parser = argparse.ArgumentParser(description=__doc__)
44
    parser.add_argument('file', type=str)
45
    args = parser.parse_args()
46
    with open(args.file) as f:
47
        data = f.read()
48
    data = re.sub(r".. raw:: html\n+\s+<div align[\s\S]*?/div>", summary, data)
49
    data = re.sub(r"Sponsors\n========[\s\S]*?Example usages", funding, data)
50
    print(data)
51

52

53
if __name__ == '__main__':
54
    main()
55