Merge pull request #43 from Technoculture/feat-executescript

implement missing `executescript` method

Author: penberg

examples/execute_script.py

@@ -0,0 +1,25 @@
+"""
+A short example showing how to execute a script containing a bunch of sql statements.
+"""
+import os
+
+import libsql_experimental as libsql
+
+def execute_script(conn, file_path: os.PathLike):
+    with open(file_path, 'r') as file:
+        script = file.read()
+
+    conn.executescript(script)
+    conn.commit()
+
+conn = libsql.connect(':memory:')
+script_path = os.path.join(os.path.dirname(__file__), 'statements.sql')
+execute_script(conn, script_path)
+
+# Retrieve the data from the 'users' table and print it
+cursor = conn.cursor()
+cursor.execute("SELECT * FROM users")
+rows = cursor.fetchall()
+print("Data in the 'users' table:")
+for row in rows:
+    print(row)

examples/statements.sql

@@ -0,0 +1,14 @@
+CREATE TABLE IF NOT EXISTS users (
+    id INTEGER PRIMARY KEY,
+    name TEXT,
+    age INTEGER
+);
+
+INSERT INTO users (name, age) VALUES ('Alice', 25);
+INSERT INTO users (name, age) VALUES ('Bob', 30);
+
+SELECT * FROM users;
+
+UPDATE users SET age = 26 WHERE name = 'Alice';
+
+SELECT * FROM users;

src/lib.rs

@@ -184,6 +184,20 @@ impl Connection {
         Ok(cursor)
     }
 
+    fn executescript(self_: PyRef<'_, Self>, script: String) -> PyResult<()> {
+        let statements = script.split(';');
+        for statement in statements {
+            let statement = statement.trim();
+            if !statement.is_empty() {
+                let cursor = Connection::cursor(&self_)?;
+                self_
+                    .rt
+                    .block_on(async { execute(&cursor, statement.to_string(), None).await })?;
+            }
+        }
+        Ok(())
+    }
+
     #[getter]
     fn isolation_level(self_: PyRef<'_, Self>) -> Option<String> {
         self_.isolation_level.clone()