2019-09-05 18:23:46 -07:00
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# 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
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
import sys
import os
import time
import notedown
import nbformat
def md2ipynb ( ) :
assert len ( sys . argv ) == 3 , ' usage: input.md output.rst '
( src_fn , input_fn , output_fn ) = sys . argv
# timeout for each notebook, in sec
timeout = 20 * 60
# if enable evaluation
do_eval = int ( os . environ . get ( ' EVAL ' , True ) )
reader = notedown . MarkdownReader ( match = ' strict ' )
with open ( input_fn , ' r ' , encoding = " utf8 " ) as f :
notebook = reader . read ( f )
if do_eval :
tic = time . time ( )
notedown . run ( notebook , timeout )
print ( ' %s : Evaluated %s in %f sec ' % ( src_fn , input_fn , time . time ( ) - tic ) )
# need to add language info to for syntax highlight
notebook [ ' metadata ' ] . update ( { ' language_info ' : { ' name ' : ' python ' } } )
2020-08-18 16:57:35 +00:00
with open ( output_fn , ' w ' , encoding = ' utf-8 ' ) as f :
2019-09-05 18:23:46 -07:00
f . write ( nbformat . writes ( notebook ) )
print ( ' %s : Write results into %s ' % ( src_fn , output_fn ) )
if __name__ == ' __main__ ' :
md2ipynb ( )
