11import os
2+ import sys
23
34
4- def rewrite_versions ():
5+ def rewrite_versions (docs_path = '/build/docs' ):
56 """
6- Collects built versions and outputs a text file that lists 'latest',
7+ Collects built versions and outputs a text file that lists 'latest',
78 latest version number, 'nightly', descending prior verions.
89 """
9- versions_txt_path = '/build/docs/ versions.txt'
10+ versions_txt_path = os . path . join ( docs_path , ' versions.txt')
1011 with open (versions_txt_path , 'w' ) as vtp :
11- vtp .write ('\n ' .join (sort_versions (os .listdir ('/build/docs' ))))
12-
12+ vtp .write ('\n ' .join (sort_versions (os .listdir (docs_path ))))
13+
1314
1415def sort_versions (versions ):
1516 v2 = versions .copy ()
@@ -18,11 +19,11 @@ def sort_versions(versions):
1819 'latest' , 'nightly' , '_sources' , 'versions.txt'
1920 } if v in v2 }
2021 if len (v2 ):
21- if len (v2 )< 2 :
22+ if len (v2 ) < 2 :
2223 return (['latest' , v2 [0 ], 'nightly' ])
23- return (['latest' , v2 [0 ], 'nightly' , * v2 [1 :] ])
24+ return (['latest' , v2 [0 ], 'nightly' ] + v2 [1 :])
2425 return (['latest' , 'nightly' ])
25-
26+
2627
2728if __name__ == '__main__' :
28- rewrite_versions ()
29+ rewrite_versions (sys . argv [ 1 ]) if len ( sys . argv ) > 1 else rewrite_versions ( )
0 commit comments