Skip to content

Instantly share code, notes, and snippets.

@mkhorasani
Last active December 26, 2022 23:18
Show Gist options
  • Star 1 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save mkhorasani/7459919bc31b3b6869a03800890dc6c3 to your computer and use it in GitHub Desktop.
Save mkhorasani/7459919bc31b3b6869a03800890dc6c3 to your computer and use it in GitHub Desktop.
def write_record(name,details,engine):
engine.execute("INSERT INTO records (name,details) VALUES ('%s','%s')" % (name,details))
def read_record(field,name,engine):
result = engine.execute("SELECT %s FROM records WHERE name = '%s'" % (field,name))
return result.first()[0]
def update_record(field,name,new_value,engine):
engine.execute("UPDATE records SET %s = '%s' WHERE name = '%s'" % (field,new_value,name))
def write_dataset(name,dataset,engine):
dataset.to_sql('%s' % (name),engine,index=False,if_exists='replace',chunksize=1000)
def read_dataset(name,engine):
try:
dataset = pd.read_sql_table(name,engine)
except:
dataset = pd.DataFrame([])
return dataset
def list_datasets(engine):
datasets = engine.execute("SELECT table_name FROM information_schema.tables WHERE table_schema = 'public' ORDER BY table_name;")
return datasets.fetchall()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment