def save(self):
with SQLite() as db:
db.execute(
'''
INSERT INTO locations
VALUES (NULL, ?)
''', (self.title, ))
return self
def find_by_email(email):
if not email:
return None
with SQLite() as db:
row = db.execute(
'SELECT * FROM user WHERE email = ?',
(email,)
).fetchone()
if row:
return User(*row)
def all(available_from=None, available_to=None):
with SQLite() as db:
rows = db.execute('SELECT * FROM post').fetchall()
if available_from=="undefined":
available_from=None
if available_to=="undefined":
available_to=None
if not available_from or not available_to:
return [Post(*row) for row in rows]
return [Post(*row) for row in rows if (
datetime.datetime.strptime(row[7], "%Y-%m-%d").date() <= datetime.datetime.strptime(available_from, "%Y-%m-%d").date() and
datetime.datetime.strptime(row[8], "%Y-%m-%d").date() >= datetime.datetime.strptime(available_to, "%Y-%m-%d").date())]
def save(self):
with SQLite() as db:
db.execute(
'''
INSERT INTO post
VALUES (NULL, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
''', (
self.title,
self.image,
self.size,
self.price,
self.bed_count,
self.location_id,
self.available_from,
self.available_to,
self.description,
self.user_id))
return self
def drop():
with SQLite() as db:
db.execute("DELETE FROM locations")
def all():
with SQLite() as db:
rows = db.execute('SELECT * FROM locations').fetchall()
return [Locations(*row) for row in rows]
def create(self):
with SQLite() as db:
db.execute('''
INSERT INTO user (email, password)
VALUES (?, ?)''', (self.email, self.password))
return self
