Enable multi-code import

minjk-bl · minjk-bl · commit 39c67a61f060 · 2022-05-06T10:29:48.000+09:00
diff --git a/js/com/component/PopupComponent.js b/js/com/component/PopupComponent.js
@@ -302,14 +302,16 @@ define([
             // Click import library
             $(this.wrapSelector('#popupImport')).on('click', function() {
                 // add import codes
-                var code = that.generateImportCode();
-                // create block and run it
-                $('#vp_wrapper').trigger({
-                    type: 'create_option_page', 
-                    blockType: 'block',
-                    menuId: 'lgExe_code',
-                    menuState: { taskState: { code: code } },
-                    afterAction: 'run'
+                var codes = that.generateImportCode();
+                codes && codes.forEach(code => {
+                    // create block and run it
+                    $('#vp_wrapper').trigger({
+                        type: 'create_option_page', 
+                        blockType: 'block',
+                        menuId: 'lgExe_code',
+                        menuState: { taskState: { code: code } },
+                        afterAction: 'run'
+                    });
                 });
             });
 
@@ -598,7 +600,7 @@ define([
 
         generateImportCode() {
             /** Implementation needed - Generated on clicking Import Library button */
-            return '';
+            return [];
         }
 
         generateCode() {
diff --git a/js/m_apps/PDF.js b/js/m_apps/PDF.js
@@ -30,6 +30,37 @@ import fitz
 import nltk
 nltk.download('punkt')`;
 
+    const PDF_FUNC = `def vp_pdf_get_sentence(fname_lst):
+    '''
+    Get sentence from pdf file by PyMuPDF
+    '''
+    df = pd.DataFrame()
+    for fname in fname_lst:
+        if fname.split('.')[-1] != 'pdf': continue
+        try:
+            doc = fitz.open(fname)
+            sentence_lst = []
+            for page in doc:
+                block_lst = page.get_text('blocks')
+        
+                text_lst = [block[4] for block in block_lst if block[6] == 0]
+                text = '\\n'.join(text_lst)
+        
+                sentence_lst.extend([sentence for sentence in nltk.sent_tokenize(text)])
+                
+            doc.close()
+        except Exception as e:
+            print(e)
+            continue
+            
+        df_doc = pd.DataFrame({
+            'fname': fname.split('/')[-1],
+            'sentence': sentence_lst
+        })
+        df = pd.concat([df,df_doc])
+        
+    return df.reset_index().drop('index', axis=1)`;
+
     const PDF_CMD = 'df = vp_pdf_get_sentence(pdf_lst)\ndf'
     /**
      * PDF
@@ -93,7 +124,10 @@ nltk.download('punkt')`;
         }
 
         generateImportCode() {
-            return PDF_IMPORT;
+            return [
+                PDF_IMPORT,
+                PDF_FUNC
+            ];
         }
 
         generateCode() {
diff --git a/js/m_apps/Profiling.js b/js/m_apps/Profiling.js
@@ -147,7 +147,9 @@ define([
         }
 
         generateImportCode() {
-            return 'from pandas_profiling import ProfileReport';
+            return [
+                'from pandas_profiling import ProfileReport'
+            ];
         }
 
         generateCode() {
diff --git a/js/m_ml/evaluation.js b/js/m_ml/evaluation.js
@@ -238,7 +238,7 @@ define([
         }
         
         generateImportCode() {
-            return 'from sklearn import metrics';
+            return ['from sklearn import metrics'];
         }
 
         generateCode() {
diff --git a/js/m_visualize/Chart.js b/js/m_visualize/Chart.js
@@ -775,7 +775,7 @@ define([
             }
             code.append("rcParams['axes.unicode_minus'] = False");
     
-            return code.toString();
+            return [code.toString()];
         }
     
 
diff --git a/js/m_visualize/ChartSetting.js b/js/m_visualize/ChartSetting.js
@@ -114,7 +114,7 @@ define([
             var code = new com_String();
             code.appendLine('import matplotlib.pyplot as plt');
             code.append('import seaborn as sns');
-            return code.toString();
+            return [code.toString()];
         }
 
         generateCode() {
diff --git a/js/m_visualize/Seaborn.js b/js/m_visualize/Seaborn.js
@@ -507,7 +507,7 @@ define([
                 code.append("rcParams['axes.unicode_minus'] = False");
             }
     
-            return code.toString();
+            return [code.toString()];
         }
 
         generateCode(preview=false) {

Original file line number	Diff line number	Diff line change
`@@ -147,7 +147,9 @@ define([`
`147`	`147`	`}`
`148`	`148`
`149`	`149`	`generateImportCode() {`
`150`		`- return 'from pandas_profiling import ProfileReport';`
	`150`	`+ return [`
	`151`	`+ 'from pandas_profiling import ProfileReport'`
	`152`	`+ ];`
`151`	`153`	`}`
`152`	`154`
`153`	`155`	`generateCode() {`
Original file line number	Diff line number	Diff line change
`@@ -238,7 +238,7 @@ define([`
`238`	`238`	`}`
`239`	`239`
`240`	`240`	`generateImportCode() {`
`241`		`- return 'from sklearn import metrics';`
	`241`	`+ return ['from sklearn import metrics'];`
`242`	`242`	`}`
`243`	`243`
`244`	`244`	`generateCode() {`
Original file line number	Diff line number	Diff line change
`@@ -775,7 +775,7 @@ define([`
`775`	`775`	`}`
`776`	`776`	`code.append("rcParams['axes.unicode_minus'] = False");`
`777`	`777`
`778`		`- return code.toString();`
	`778`	`+ return [code.toString()];`
`779`	`779`	`}`
`780`	`780`
`781`	`781`
Original file line number	Diff line number	Diff line change
`@@ -114,7 +114,7 @@ define([`
`114`	`114`	`var code = new com_String();`
`115`	`115`	`code.appendLine('import matplotlib.pyplot as plt');`
`116`	`116`	`code.append('import seaborn as sns');`
`117`		`- return code.toString();`
	`117`	`+ return [code.toString()];`
`118`	`118`	`}`
`119`	`119`
`120`	`120`	`generateCode() {`
Original file line number	Diff line number	Diff line change
`@@ -507,7 +507,7 @@ define([`
`507`	`507`	`code.append("rcParams['axes.unicode_minus'] = False");`
`508`	`508`	`}`
`509`	`509`
`510`		`- return code.toString();`
	`510`	`+ return [code.toString()];`
`511`	`511`	`}`
`512`	`512`
`513`	`513`	`generateCode(preview=false) {`