Merge pull request #233 from swgillespie/master

Add support for publishing to PyPI #203

pkg/plugin/publish/publish.go

@@ -10,6 +10,7 @@ import (
 type Publish struct {
 	S3    *S3    `yaml:"s3,omitempty"`
 	Swift *Swift `yaml:"swift,omitempty"`
+	PyPI  *PyPI  `yaml:"pypi,omitempty"`
 }
 
 func (p *Publish) Write(f *buildfile.Buildfile) {
@@ -19,4 +20,7 @@ func (p *Publish) Write(f *buildfile.Buildfile) {
 	if p.Swift != nil {
 		p.Swift.Write(f)
 	}
+	if p.PyPI != nil {
+		p.PyPI.Write(f)
+	}
 }

pkg/plugin/publish/pypi.go

@@ -1 +1,68 @@
 package publish
+
+import (
+	"fmt"
+	"github.com/drone/drone/pkg/build/buildfile"
+)
+
+// set up the .pypirc file
+var pypirc = `
+cat <<EOF > $HOME/.pypirc
+[distutils]
+index-servers = 
+    pypi
+
+[pypi]
+username:%s
+password:%s
+EOF`
+
+var deployCmd = `
+if [ -z $_PYPI_SETUP_PY ]
+then
+    python $_PYPI_SETUP_PY sdist %s upload
+    if [ $? -ne 0 ]
+    then
+        echo "Deploy to PyPI failed - perhaps due to the version number not being incremented. Continuing..."
+    fi
+else
+    echo "Failed to find setup.py file"
+fi
+`
+
+type PyPI struct {
+	Username string   `yaml:"username,omitempty"`
+	Password string   `yaml:"password,omitempty"`
+	Formats  []string `yaml:"formats,omitempty"`
+}
+
+func (p *PyPI) Write(f *buildfile.Buildfile) {
+	if len(p.Username) == 0 || len(p.Password) == 0 {
+		// nothing to do if the config is fundamentally flawed
+		return
+	}
+	f.WriteCmdSilent("echo 'publishing to PyPI...'")
+
+	// find the setup.py file
+	f.WriteCmdSilent("_PYPI_SETUP_PY=$(find . -name 'setup.py')")
+
+	// build the .pypirc file that pypi expects
+	f.WriteCmdSilent(fmt.Sprintf(pypirc, p.Username, p.Password))
+	formatStr := p.BuildFormatStr()
+
+	// if we found the setup.py file use it to deploy
+	f.WriteCmdSilent(fmt.Sprintf(deployCmd, formatStr))
+}
+
+func (p *PyPI) BuildFormatStr() string {
+	if len(p.Formats) == 0 {
+		// the format parameter is optional - if it's not here,
+		// omit the format string completely.
+		return ""
+	}
+	fmtStr := "--formats "
+	for i := range p.Formats {
+		fmtStr += p.Formats[i] + ","
+	}
+	return fmtStr[:len(fmtStr)-1]
+}