summaryrefslogtreecommitdiff
path: root/setup.py
blob: 132a8ef00d03436e0ce839e36ad73091d46765b8 (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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
# Copyright (C) 2012-2015, 2019 Codethink Limited
#
# This program 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; version 2 of the License.
#
# This program 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 this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.


"""Setup.py for lorry."""


import compileall
import distutils.dir_util
import glob
import os
import shutil
import subprocess
import sys
from distutils.cmd import Command
from distutils.command.build import build
from distutils.command.clean import clean
from distutils.command.install import install
from distutils.core import setup


class Build(build):
    def run(self):
        build.run(self)

        compileall.compile_dir("hg-fast-export/plugins")

        print("building manpages")
        for x in ["lorry"]:
            with open("%s.1" % x, "w") as f:
                subprocess.check_call(
                    [
                        "python3",
                        x,
                        "--generate-manpage=%s.1.in" % x,
                        "--output=%s.1" % x,
                    ],
                    stdout=f,
                )


class Clean(clean):
    clean_files = [".coverage", "build", "lorry.1"]
    clean_globs = [
        "*/*.py[co]",
        "hg-fast-export/plugins/*/__pycache__",
    ]

    def run(self):
        clean.run(self)
        itemses = [self.clean_files] + [glob.glob(x) for x in self.clean_globs]
        for items in itemses:
            for filename in items:
                if os.path.isdir(filename):
                    shutil.rmtree(filename)
                elif os.path.exists(filename):
                    os.remove(filename)


class Check(Command):
    user_options = []

    def initialize_options(self):
        pass

    def finalize_options(self):
        pass

    def run(self):
        subprocess.check_call(["./check"])


class Install(install):
    def run(self):
        install.run(self)

        # Install hg-fast-export plugins in a private directory
        distutils.dir_util.copy_tree(
            "hg-fast-export/plugins",
            os.path.join(self.install_data, "share/lorry/hg-fast-export/plugins"),
        )


if sys.version_info < (3, 6):
    print("lorry requires Python 3.6+", file=sys.stderr)
    sys.exit(1)

setup(
    name="lorry",
    description="FIXME",
    long_description="""\
FIXME
""",
    author="Baserock",
    author_email="baserock-dev@baserock.org",
    url="http://wiki.baserock.org/",
    scripts=[
        "lorry",
        "lorry.tar-importer",
        "lorry.gzip-importer",
        "lorry.zip-importer",
        "lorry-ssh-wrapper",
        "lorry.raw-file-importer",
    ],
    data_files=[("share/man/man1", glob.glob("*.[1-8]"))],
    cmdclass={
        "build": Build,
        "check": Check,
        "clean": Clean,
        "install": Install,
    },
)