with open("CHANGES.rst") as f:

contents = f.read()

toks = re.split("\-{3,}", contents)

n = len(toks[0].split()[-1])

changes = [toks[0]]

changes.append("\n" + "\n".join(toks[1].strip().split("\n")[0:-1]))

changes = ("-" * n).join(changes)

with open("docs_rst/latest_changes.rst", "w") as f:

f.write(changes)

with cd("docs_rst"):

ctx.run("cp ../CHANGES.rst change_log.rst")

ctx.run("rm pymatgen.*.rst")

ctx.run("sphinx-apidoc --separate -d 7 -o . -f ../pymatgen")

ctx.run("rm pymatgen*.tests.*rst")

for f in glob.glob("*.rst"):

if f.startswith('pymatgen') and f.endswith('rst'):

newoutput = []

suboutput = []

subpackage = False

with open(f, 'r') as fid:

for line in fid:

clean = line.strip()

if clean == "Subpackages":

subpackage = True

if not subpackage and not clean.endswith("tests"):

newoutput.append(line)

else:

if not clean.endswith("tests"):

suboutput.append(line)

if clean.startswith("pymatgen") and not clean.endswith("tests"):

newoutput.extend(suboutput)

subpackage = False

suboutput = []

with open(f, 'w') as fid:

fid.write("".join(newoutput))

ctx.run("make html")

ctx.run("cp _static/* ../docs/html/_static")

with cd("docs"):

ctx.run("rm *.html")

ctx.run("cp -r html/* .")

ctx.run("rm -r html")

ctx.run("rm -r doctrees")

ctx.run("rm -r _sources")

ctx.run("rm -r _build", warn=True)

# This makes sure pymatgen.org works to redirect to the Gihub page

ctx.run("echo \"pymatgen.org\" > CNAME")

# Avoid the use of jekyll so that _dir works as intended.

ctx.run("cp docs_rst/conf-docset.py docs_rst/conf.py")

make_doc(ctx)

ctx.run('rm docs/_static/pymatgen.docset.tgz', warn=True)

ctx.run('doc2dash docs -n pymatgen -i docs/_images/pymatgen.png -u https://pymatgen.org/')

plist = "pymatgen.docset/Contents/Info.plist"

xml = []

with open(plist, "rt") as f:

for l in f:

xml.append(l.strip())

if l.strip() == "<dict>":

xml.append("<key>dashIndexFilePath</key>")

xml.append("<string>index.html</string>")

with open(plist, "wt") as f:

f.write("\n".join(xml))

ctx.run('tar --exclude=".DS_Store" -cvzf pymatgen.tgz pymatgen.docset')

xml = []

with open("docs/pymatgen.xml") as f:

for l in f:

l = l.strip()

if l.startswith("<version>"):

xml.append("<version>%s</version>" % NEW_VER)

else:

xml.append(l)

with open("docs/pymatgen.xml", "wt") as f:

f.write("\n".join(xml))

make_dash(ctx)

ctx.run('cp pymatgen.tgz ../Dash-User-Contributions/docsets/pymatgen/pymatgen.tgz')

with cd("../Dash-User-Contributions/docsets/pymatgen"):

with open("docset.json", "rt") as f:

data = json.load(f)

data["version"] = NEW_VER

with open("docset.json", "wt") as f:

json.dump(data, f, indent=4)

ctx.run('git commit -a -m "Update to v%s"' % NEW_VER)

ctx.run('git push')

with cd("../Dash-User-Contributions/docsets/pymatgen"):

payload = {

"title": "Update pymatgen docset to v%s" % NEW_VER,

"body": "Update pymatgen docset to v%s" % NEW_VER,

"head": "Dash-User-Contributions:master",

"base": "master"

}

response = requests.post(

"https://api.github.com/repos/materialsvirtuallab/Dash-User-Contributions/pulls",

data=json.dumps(payload))

make_doc(ctx)

try:

contribute_dash(ctx)

except:

pass

ctx.run("cp docs_rst/conf-normal.py docs_rst/conf.py")

make_doc(ctx)

ctx.run("git add .")

ctx.run("git commit -a -m \"Update docs\"")

lines = []

with open("pymatgen/__init__.py", "rt") as f:

for l in f:

if "__version__" in l:

lines.append('__version__ = "%s"' % NEW_VER)

else:

lines.append(l.rstrip())

with open("pymatgen/__init__.py", "wt") as f:

f.write("\n".join(lines))

lines = []

with open("setup.py", "rt") as f:

for l in f:

lines.append(re.sub(r'version=([^,]+),', 'version="%s",' % NEW_VER,

l.rstrip()))

with open("setup.py", "wt") as f:

with cd("docs/_build/html/"):

ctx.run("git pull")

ctx.run("nosetests --config=nose.cfg --cover-html --cover-html-dir=docs/_build/html/coverage")

ctx.run("git tag -a v%s -m \"v%s release\"" % (NEW_VER, NEW_VER))

ctx.run("git push --tags")

ctx.run("git checkout stable")

ctx.run("git pull")

ctx.run("git merge master")

ctx.run("git push")

with open("CHANGES.rst") as f:

contents = f.read()

toks = re.split(r"\-+", contents)

desc = toks[1].strip()

toks = desc.split("\n")

desc = "\n".join(toks[:-1]).strip()

payload = {

"tag_name": "v" + NEW_VER,

"target_commitish": "master",

"name": "v" + NEW_VER,

"body": desc,

"draft": False,

"prerelease": False

}

response = requests.post(

"https://api.github.com/repos/materialsproject/pymatgen/releases",

data=json.dumps(payload),

headers={"Authorization": "token " + os.environ["GITHUB_RELEASES_TOKEN"]})

with open("CHANGES.rst") as f:

contents = f.read()

toks = re.split(r"\-+", contents)

desc = toks[1].strip()

toks = desc.split("\n")

desc = "\n".join(toks[:-1]).strip()

raw = "v" + NEW_VER + "\n\n" + desc

payload = {

"topic_id": 36,

"raw": raw,

}

response = requests.post(

"https://pymatgen.discourse.group/posts.json",

data=payload,

params={

"api_username": os.environ["DISCOURSE_API_USERNAME"],

"api_key": os.environ["DISCOURSE_API_KEY"]}

)

output = subprocess.check_output(["git", "log", "--pretty=format:%s",

"v%s..HEAD" % CURRENT_VER])

lines = ["* " + l for l in output.decode("utf-8").strip().split("\n")]

with open("CHANGES.rst") as f:

contents = f.read()

l = "=========="

toks = contents.split(l)

head = "\n\nv%s\n" % NEW_VER + "-" * (len(NEW_VER) + 1) + "\n"

toks.insert(-1, head + "\n".join(lines))

with open("CHANGES.rst", "w") as f:

f.write(toks[0] + l + "".join(toks[1:]))

ctx.run("rm -r dist build pymatgen.egg-info", warn=True)

set_ver(ctx)

if not notest:

ctx.run("nosetests")

publish(ctx)

if not nodoc:

update_doc(ctx)

merge_stable(ctx)

release_github(ctx)