Spaces:
Running
on
Zero
Running
on
Zero
import gradio as gr | |
#from mysite.libs.utilities import chat_with_interpreter, completion, process_file,no_process_file | |
from interpreter import interpreter | |
import mysite.interpreter.interpreter_config # インポートするだけで設定が適用されます | |
import duckdb | |
import gradio as gr | |
import psycopg2 | |
from dataclasses import dataclass, field | |
from typing import List, Optional | |
import os | |
#from mysite.interpreter.process import no_process_file,process_file | |
#from controllers.gra_04_database.rides import test_set_lide | |
class Ride: | |
ride_id: Optional[int] = field(default=None) | |
rideable_type: str = '' | |
start_station_id: int = 0 | |
start_station_name: str = '' | |
end_station_id: int = 0 | |
end_station_name: str = '' | |
started_at: str = '' | |
ended_at: str = '' | |
member_casual: str = '' | |
def connect_to_db(): | |
conn = psycopg2.connect( | |
dbname="neondb", | |
user=os.getenv("postgre_user"), | |
password=os.getenv("postgre_pass"), | |
host=os.getenv("postgre_host"), | |
port=5432, | |
sslmode="require" | |
) | |
return conn | |
def create_ride(ride: Ride): | |
conn = connect_to_db() | |
cur = conn.cursor() | |
cur.execute("INSERT INTO rides (rideable_type, start_station_id, start_station_name, end_station_id, end_station_name, started_at, ended_at, member_casual) VALUES (%s, %s, %s, %s, %s, %s, %s, %s) RETURNING ride_id", | |
(ride.rideable_type, ride.start_station_id, ride.start_station_name, ride.end_station_id, ride.end_station_name, ride.started_at, ride.ended_at, ride.member_casual)) | |
ride_id = cur.fetchone()[0] | |
conn.commit() | |
cur.close() | |
conn.close() | |
return ride_id | |
def read_rides(): | |
conn = connect_to_db() | |
cur = conn.cursor() | |
cur.execute("SELECT * FROM rides") | |
rides = cur.fetchall() | |
conn.close() | |
return rides | |
def read_ride(ride_id: int): | |
conn = connect_to_db() | |
cur = conn.cursor() | |
cur.execute("SELECT * FROM rides WHERE ride_id = %s", (ride_id,)) | |
ride = cur.fetchone() | |
conn.close() | |
return ride | |
def update_ride(ride: Ride): | |
conn = connect_to_db() | |
cur = conn.cursor() | |
no_process_file(ride.start_station_name,ride.end_station_name) | |
cur.execute("UPDATE rides SET rideable_type = %s, start_station_id = %s, start_station_name = %s, end_station_id = %s, end_station_name = %s, started_at = %s, ended_at = %s, member_casual = %s WHERE ride_id = %s", | |
(ride.rideable_type, ride.start_station_id, ride.start_station_name, ride.end_station_id, ride.end_station_name, ride.started_at, ride.ended_at, ride.member_casual, ride.ride_id)) | |
conn.commit() | |
cur.close() | |
conn.close() | |
def delete_ride(ride_id: int): | |
conn = connect_to_db() | |
cur = conn.cursor() | |
cur.execute("DELETE FROM rides WHERE ride_id = %s", (ride_id,)) | |
conn.commit() | |
cur.close() | |
conn.close() | |
def test_set_lide(input="test",foldername="test"): | |
ride = Ride( | |
rideable_type="rideable_type", | |
start_station_name=input, | |
end_station_name=foldername, | |
) | |
create_ride(ride) | |
return [[r[0], r[1], r[2], r[3], r[4], r[5], r[6], r[7], r[8]] for r in read_rides()] |