~james-w/lpdistutils/trunk

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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
"""distutils.command.upload

Implements the Distutils 'upload' subcommand (upload package to PyPI)."""

import datetime
from distutils.errors import DistutilsOptionError
from distutils.spawn import spawn
import os
import platform
import sys
import subprocess
import tempfile

from lpdistutils import LPCommand


class lpupload(LPCommand):

    description = "upload files to Launchpad"
    user_options = LPCommand.user_options + [
        ('sign', 's',
         'sign files to upload using gpg'),
        ('identity=', 'i', 'GPG identity used to sign files'),
        ]
    boolean_options = LPCommand.boolean_options + ['sign']

    def initialize_options(self):
        LPCommand.initialize_options(self)
        self.sign = False
        self.identity = None

    def finalize_options(self):
        LPCommand.finalize_options(self)
        if self.identity and not self.sign:
            raise DistutilsOptionError(
                "Must use --sign for --identity to have meaning"
            )

    def run(self):
        if not self.distribution.dist_files:
            raise DistutilsOptionError("No dist file created in earlier command")
        for command, pyversion, filename in self.distribution.dist_files:
            self.upload_file(command, pyversion, filename)

    def create_release(self, project, version):
        print "Release %s could not be found for project. Create it? (Y/n)" % version
        answer = sys.stdin.readline().strip().lower()
        if not answer.startswith('y'):
            sys.exit(0)
        milestone = None
        if len(project.series) != 1:
            eligble_series = []
            found = False
            for s in project.series:
                for m in s.all_milestones:
                    if m.name == version:
                        milestone = m
                        series = s
                        found = True
                        break
                if found:
                    break
            if series is None:
                print "Select a series to upload to:"
                for i, series in enumerate(project.series):
                    print "%d: %s" % (i+1, series.name)
                answer = int(sys.stdin.readline().strip())
                if answer < 1 or answer > len(series):
                    sys.exit(0)
                series = project.series[answer-1]
        else:
            series = project.series[0]

        release_date = datetime.date.today().strftime('%Y-%m-%d')
        if milestone is None:
            milestone = series.newMilestone(name=version,
                    date_targeted=release_date)
        return milestone.createProductRelease(date_released=release_date)

    def edit_file(self, prefix, description):
        (fd, f) = tempfile.mkstemp(prefix=prefix+'.')
        os.write(fd, '\n\n#------\n# Please enter the %s here. Lines which start with "#" are ignored.\n' %
                description)
        os.close(fd)
        subprocess.call(['sensible-editor', f])
        content = ''
        for l in open(f):
            if l.startswith('#'):
                continue
            content += l

        return content.strip()

    def upload_file(self, command, pyversion, filename):
        # Sign if requested
        if self.sign:
            gpg_args = ["gpg", "--detach-sign", "-a", filename]
            if self.identity:
                gpg_args[2:2] = ["--local-user", self.identity]
            spawn(gpg_args)

        # Fill in the data
        f = open(filename,'rb')
        content = f.read()
        f.close()
        basename = os.path.basename(filename)
        description = command + " (%s)" % str(pyversion)
        if command=='bdist_egg' and self.distribution.has_ext_modules():
            description += " (built on %s)" % platform.platform(terse=1)
        if command == 'bdist_rpm':
            dist, version, id = platform.dist()
            if dist:
                description += ' (built for %s %s)' % (dist, version)
        elif command == 'bdist_dumb':
            description += ' (built for %s)' % platform.platform(terse=1)

        lpinfo = self.get_lpinfo()
        name = lpinfo.get_name()
        if name is None:
            name = self.distribution.get_name()
        version = self.distribution.metadata.version

        lp = self.get_lp()
        try:
            project = lp.projects[name]
        except KeyError:
            raise KeyError("Project %s not found." % name)

        release = None
        for rel in project.releases:
            if rel.version == version:
                release = rel
                break
        if release is None:
            release = self.create_release(project, version)

        kwargs = {}
        sig_filename = filename + ".asc"
        if self.sign:
            kwargs["signature_filename"] = os.path.basename(sig_filename)
            kwargs["signature_content"] = open(sig_filename).read()

        release.add_file(filename=basename, description=description,
                file_content=content, file_type="Code Release Tarball",
                content_type='application/x-gzip', **kwargs)

        changelog = self.edit_file('changelog', 'changelog')
        if changelog:
            release.changelog = changelog
        release_notes = self.edit_file('releasenotes', 'release notes')
        if release_notes:
            release.release_notes = release_notes

        release.lp_save()