1 files changed, 25 insertions, 1 deletions
diff --git a/src/main/Analyzor.java b/src/main/Analyzor.java
index 639f385..754e593 100644
--- a/src/main/Analyzor.java
+++ b/src/main/Analyzor.java
@@ -5,6 +5,8 @@ import database.QueryUtils;
 import java.io.File;
 import java.io.FileNotFoundException;
 import java.io.IOException;
+import java.io.PrintWriter;
+import java.io.UnsupportedEncodingException;
 import java.sql.Connection;
 import java.sql.PreparedStatement;
 import java.sql.ResultSet;
@@ -172,7 +174,29 @@ public class Analyzor {
             }
         }
     }
-
+    
+    //generate csv for disco from the query
+    public void disco(String query) throws SQLException, FileNotFoundException, UnsupportedEncodingException{
+        query(query);
+        PrintWriter writer = new PrintWriter("output.csv", "UTF-8");
+        while(data.next()){
+            for(int i = 1; i<data.getMetaData().getColumnCount();i++){
+                if(data.getObject(i)==null){
+                    writer.print(", ");
+                } else {
+                    writer.print(data.getObject(i).toString().replaceAll("[,\n]", " ")+", ");
+                }
+            }
+            if(data.getObject(data.getMetaData().getColumnCount())==null){
+                writer.println(" ");
+            } else {
+                writer.println(data.getObject(data.getMetaData().getColumnCount()).toString().replace(",", " "));
+            }
+        }
+        writer.close();
+    }
+    
+    
     //replaces punctuation so it will be splitted
     //also removes urls
     private String splitPunctToWords(String text) {