From 15d0ce3123833c53c979c9f2aa0ac7dd724c192f Mon Sep 17 00:00:00 2001
From: "Uwe L. Korn" <uwelk@xhochy.com>
Date: Wed, 19 Oct 2016 10:06:09 +0200
Subject: [PATCH] ARROW-342: Set Python version on release

Change-Id: I091ca7d9b4523608ad51491b25e5a66cc85c84d8
---
 dev/release/00-prepare.sh  |  9 +++++++--
 python/.gitignore          |  1 +
 python/pyarrow/__init__.py |  1 +
 python/setup.py            | 24 ++++++++++++++++++++----
 4 files changed, 29 insertions(+), 6 deletions(-)

diff --git a/dev/release/00-prepare.sh b/dev/release/00-prepare.sh
index 3c1fb9a0938..3423a3e6c5b 100644
--- a/dev/release/00-prepare.sh
+++ b/dev/release/00-prepare.sh
@@ -7,9 +7,9 @@
 # to you under the Apache License, Version 2.0 (the
 # "License"); you may not use this file except in compliance
 # with the License.  You may obtain a copy of the License at
-# 
+#
 #   http://www.apache.org/licenses/LICENSE-2.0
-# 
+#
 # Unless required by applicable law or agreed to in writing,
 # software distributed under the License is distributed on an
 # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
@@ -43,4 +43,9 @@ mvn release:prepare -Dtag=${tag} -DreleaseVersion=${version} -DautoVersionSubmod
 
 cd -
 
+cd "${SOURCE_DIR}/../../python"
+sed -i "s/VERSION = '[^']*'/VERSION = '${version}'/g" setup.py
+sed -i "s/ISRELEASED = False/ISRELEASED = True/g" setup.py
+cd -
+
 echo "Finish staging binary artifacts by running: sh dev/release/01-perform.sh"
diff --git a/python/.gitignore b/python/.gitignore
index 7e2e360557a..07f28355a25 100644
--- a/python/.gitignore
+++ b/python/.gitignore
@@ -25,6 +25,7 @@ MANIFEST
 # Generated sources
 *.c
 *.cpp
+pyarrow/version.py
 # Python files
 
 # setup.py working directory
diff --git a/python/pyarrow/__init__.py b/python/pyarrow/__init__.py
index 8b131aaa8f4..775ce7ec475 100644
--- a/python/pyarrow/__init__.py
+++ b/python/pyarrow/__init__.py
@@ -42,3 +42,4 @@
                             DataType, Field, Schema, schema)
 
 from pyarrow.table import Column, RecordBatch, Table, from_pandas_dataframe
+from pyarrow.version import version as __version__
diff --git a/python/setup.py b/python/setup.py
index d040ea7e892..99049777514 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -50,10 +50,25 @@
 if Cython.__version__ < '0.19.1':
     raise Exception('Please upgrade to Cython 0.19.1 or newer')
 
-MAJOR = 0
-MINOR = 1
-MICRO = 0
-VERSION = '%d.%d.%ddev' % (MAJOR, MINOR, MICRO)
+VERSION = '0.1.0'
+ISRELEASED = False
+
+if not ISRELEASED:
+    VERSION += '.dev'
+
+setup_dir = os.path.abspath(os.path.dirname(__file__))
+
+
+def write_version_py(filename=os.path.join(setup_dir, 'pyarrow/version.py')):
+    a = open(filename, 'w')
+    file_content = "\n".join(["",
+                              "# THIS FILE IS GENERATED FROM SETUP.PY",
+                              "version = '%(version)s'",
+                              "isrelease = '%(isrelease)s'"])
+
+    a.write(file_content % {'version': VERSION,
+                            'isrelease': str(ISRELEASED)})
+    a.close()
 
 
 class clean(_clean):
@@ -238,6 +253,7 @@ def get_outputs(self):
         return [self._get_cmake_ext_path(name)
                 for name in self.get_names()]
 
+write_version_py()
 
 DESC = """\
 Python library for Apache Arrow"""