Ticket #10637: add_sws2rst.patch

sage-sage

@@ -144 +144 @@
     echo "  -fixdoctests <file.py> -- create <file.py>.out that would pass the doctests"
     echo "                            and output a patch"
     echo "  -startuptime        -- display how long each component of Sage takes to"
+    echo "  -sws2rst <sws document>  Generates a rst file from the given document"
     echo "                         start up"
     echo "  -t [options] <files|dir>"
     echo "                      -- test examples in .py, .pyx, .sage or .tex files"
@@ -700 +701 @@
    echo -n " Do you want to proceed? [y/n] "
    read CHOICE
    while [ "$CHOICE" != "y" -a "$CHOICE" != "n" ]; do
-           echo -n " Choose one of y, n: "
-           read CHOICE
+      echo -n " Choose one of y, n: "
+      read CHOICE
    done
    if [ $CHOICE = 'n' ]; then
-           exit 0
+       exit 0
    fi
    build_sage -b "$@"
    exit $?
@@ -804 +805 @@
        PKG_NAME=`echo "$PKG" | sed -e "s/\.spkg$//"`
        PKG_NAME=`basename "$PKG_NAME"`
        case $PKG in
-           /*)
-               sage-spkg $OPT "$PKG" 2>&1 | (trap "" SIGINT; tee -a ../install.log "$SAGE_LOGS/$PKG_NAME".log)
-           ;;
-           *)
-               sage-spkg $OPT "$CUR/$PKG" 2>&1 | (trap "" SIGINT; tee -a ../install.log "$SAGE_LOGS/$PKG_NAME".log)
-           ;;
+       /*)
+           sage-spkg $OPT "$PKG" 2>&1 | (trap "" SIGINT; tee -a ../install.log "$SAGE_LOGS/$PKG_NAME".log)
+       ;;
+       *)
+           sage-spkg $OPT "$CUR/$PKG" 2>&1 | (trap "" SIGINT; tee -a ../install.log "$SAGE_LOGS/$PKG_NAME".log)
+       ;;
        esac
 
        if [ $? -ne 0 ]; then
@@ -1008 +1009 @@
     python "$SAGE_LOCAL"/bin/sage-startuptime.py
     exit $?
 fi
+
+if [ "$1" = '-sws2rst' -o "$1" = '--sws2rst' ]; then
+   cd "$CUR"
+   echo hola, "$CUR"
+   shift
+   sage-sws2rst "$@"
+   exit $?
+fi
+
 if [ "$1" = '-gthread' -o "$1" = '-qthread' -o "$1" = '-q4thread' -o "$1" = '-wthread' -o "$1" = '-pylab' ]; then
     sage "$1"
     exit $?

new file sage-sws2rst

@@ +1 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+import sys
+import tarfile
+import os
+import shutil
+import codecs
+from sagenb.misc.worksheet2rst import parse
+
+if __name__=='__main__':
+    if len(sys.argv)<=1:
+        print 'First argument should be a sws file'
+        sys.exit()
+
+#    retcode = call(['tar','jxf', sys.argv[1], '-o',path2])
+    
+    fichero_sws = tarfile.open(sys.argv[1], mode='r:bz2')
+    #TODO: python complains about using tempnam, but I don't
+    #know hot to fix it or see any danger
+    tempname = os.tempnam('.')
+    fichero_sws.extractall(tempname)
+    nombre_base = os.path.splitext(sys.argv[1])[0]
+
+    #Imagenes
+    images_dir = nombre_base + '_media'
+    if not os.path.exists(images_dir):
+        os.mkdir(images_dir)
+    
+    #"data" dir
+    data_path = os.path.join(tempname,'sage_worksheet','data')
+    if os.path.exists(data_path):
+        for image in os.listdir(data_path):
+            try:
+                shutil.move(os.path.join(data_path, image), images_dir)
+            except shutil.Error:
+                pass
+
+    #cells
+    cells_path = os.path.join(tempname,'sage_worksheet','cells')    
+    for cell in os.listdir(cells_path):
+        cell_path = os.path.join(cells_path, cell)
+        for image in os.listdir(cell_path):
+            shutil.copy2(os.path.join(cell_path, image), images_dir)
+            shutil.move(os.path.join(images_dir, image),
+                        os.path.join(images_dir, 'cell_%s_%s'%(cell,image)))
+    
+    #Fichero rst
+    fichero = codecs.open(os.path.join('.',tempname,'sage_worksheet','worksheet.html'),
+                          mode='r',
+                          encoding='utf-8')
+    html_text = fichero.read()
+    fichero.close()
+    rst_text = parse(html_text, images_dir = images_dir)
+    fichero_rst = nombre_base + '.rst'
+    fichero_out = codecs.open(fichero_rst, mode='w',
+                  encoding='utf-8')
+    fichero_out.write(rst_text)
+    fichero_out.close()
+
+    shutil.rmtree(tempname)