import csv, sqlite3 # Will use these to dump database contents to csv file
import os
from models import (DB, Buyer, Supplier, Product, Staff, Courier, Unit, Descriptor,
Brand, Stock, Order)
from peewee import *
try:
os.mkdir('csv')
except FileExistsError:
pass
db = sqlite3.connect('peewee.db')
for table in DB.get_tables():
with open('csv/{}s.csv'.format(table), 'w') as csv_file:
try :
writer = csv.writer(csv_file)
cur = db.cursor()
cur.execute('SELECT * FROM {}'.format(table))
writer.writerow([description[0] for description in cur.description])
writer.writerows(cur)
except sqlite3.OperationalError:
pass
print('Dump taken with much success')