e8c28a4a91
You should delete your database and let the software recreate it after this commit. Alternatively, add the `is_expired` column to the flats table.
217 lines
5.8 KiB
Python
217 lines
5.8 KiB
Python
# coding: utf-8
|
|
"""
|
|
This modules defines an SQLAlchemy ORM model for a flat.
|
|
"""
|
|
# pylint: disable=locally-disabled,invalid-name,too-few-public-methods
|
|
from __future__ import absolute_import, print_function, unicode_literals
|
|
|
|
import logging
|
|
import enum
|
|
|
|
import arrow
|
|
|
|
from sqlalchemy import (
|
|
Boolean, Column, DateTime, Enum, Float, SmallInteger, String, Text, inspect
|
|
)
|
|
from sqlalchemy.orm import validates
|
|
|
|
from flatisfy.database.base import BASE
|
|
from flatisfy.database.types import MagicJSON
|
|
|
|
|
|
LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class FlatUtilities(enum.Enum):
|
|
"""
|
|
An enum of the possible utilities status for a flat entry.
|
|
"""
|
|
included = 10
|
|
unknown = 0
|
|
excluded = -10
|
|
|
|
|
|
class FlatStatus(enum.Enum):
|
|
"""
|
|
An enum of the possible status for a flat entry.
|
|
"""
|
|
user_deleted = -100
|
|
duplicate = -20
|
|
ignored = -10
|
|
new = 0
|
|
followed = 10
|
|
contacted = 20
|
|
answer_no = 30
|
|
answer_yes = 31
|
|
|
|
|
|
# List of statuses that are automatically handled, and which the user cannot
|
|
# manually set through the UI.
|
|
AUTOMATED_STATUSES = [
|
|
FlatStatus.new,
|
|
FlatStatus.duplicate,
|
|
FlatStatus.ignored
|
|
]
|
|
|
|
|
|
class Flat(BASE):
|
|
"""
|
|
SQLAlchemy ORM model to store a flat.
|
|
"""
|
|
__tablename__ = "flats"
|
|
__searchable__ = [
|
|
"title", "text", "station", "location", "details", "notes"
|
|
]
|
|
|
|
# Weboob data
|
|
id = Column(String, primary_key=True)
|
|
area = Column(Float)
|
|
bedrooms = Column(Float)
|
|
cost = Column(Float)
|
|
currency = Column(String)
|
|
utilities = Column(Enum(FlatUtilities), default=FlatUtilities.unknown)
|
|
date = Column(DateTime)
|
|
details = Column(MagicJSON)
|
|
location = Column(String)
|
|
phone = Column(String)
|
|
photos = Column(MagicJSON)
|
|
rooms = Column(Float)
|
|
station = Column(String)
|
|
text = Column(Text)
|
|
title = Column(String)
|
|
urls = Column(MagicJSON)
|
|
merged_ids = Column(MagicJSON)
|
|
notes = Column(Text)
|
|
notation = Column(SmallInteger, default=0)
|
|
is_expired = Column(Boolean, default=False)
|
|
|
|
# Flatisfy data
|
|
# TODO: Should be in another table with relationships
|
|
flatisfy_stations = Column(MagicJSON)
|
|
flatisfy_postal_code = Column(String)
|
|
flatisfy_time_to = Column(MagicJSON)
|
|
flatisfy_constraint = Column(String)
|
|
|
|
# Status
|
|
status = Column(Enum(FlatStatus), default=FlatStatus.new)
|
|
|
|
# Date for visit
|
|
visit_date = Column(DateTime)
|
|
|
|
@validates('utilities')
|
|
def validate_utilities(self, _, utilities):
|
|
"""
|
|
Utilities validation method
|
|
"""
|
|
if isinstance(utilities, FlatUtilities):
|
|
return utilities
|
|
|
|
if utilities == "C.C.":
|
|
return FlatUtilities.included
|
|
elif utilities == "H.C.":
|
|
return FlatUtilities.excluded
|
|
return FlatUtilities.unknown
|
|
|
|
@validates("status")
|
|
def validate_status(self, _, status):
|
|
"""
|
|
Status validation method
|
|
"""
|
|
if isinstance(status, FlatStatus):
|
|
return status
|
|
|
|
try:
|
|
return getattr(FlatStatus, status)
|
|
except (AttributeError, TypeError):
|
|
LOGGER.warn("Unkown flat status %s, ignoring it.",
|
|
status)
|
|
return self.status.default.arg
|
|
|
|
@validates("notation")
|
|
def validate_notation(self, _, notation):
|
|
"""
|
|
Notation validation method
|
|
"""
|
|
try:
|
|
notation = int(notation)
|
|
assert notation >= 0 and notation <= 5
|
|
except (ValueError, AssertionError):
|
|
raise ValueError('notation should be an integer between 0 and 5')
|
|
return notation
|
|
|
|
@validates("date")
|
|
def validate_date(self, _, date):
|
|
"""
|
|
Date validation method
|
|
"""
|
|
return arrow.get(date).naive
|
|
|
|
@validates("visit_date")
|
|
def validate_visit_date(self, _, visit_date):
|
|
"""
|
|
Visit date validation method
|
|
"""
|
|
return arrow.get(visit_date).naive
|
|
|
|
@validates("photos")
|
|
def validate_photos(self, _, photos):
|
|
"""
|
|
Photos validation method
|
|
"""
|
|
if not photos:
|
|
photos = []
|
|
for photo in photos:
|
|
try:
|
|
# Remove computed hash to avoid storing it in db
|
|
del photo["hash"]
|
|
except KeyError:
|
|
pass
|
|
return photos
|
|
|
|
@staticmethod
|
|
def from_dict(flat_dict):
|
|
"""
|
|
Create a Flat object from a flat dict as manipulated by the filtering
|
|
pass.
|
|
"""
|
|
# Handle flatisfy metadata
|
|
flat_dict = flat_dict.copy()
|
|
if "flatisfy" in flat_dict:
|
|
flat_dict["flatisfy_stations"] = (
|
|
flat_dict["flatisfy"].get("matched_stations", [])
|
|
)
|
|
flat_dict["flatisfy_postal_code"] = (
|
|
flat_dict["flatisfy"].get("postal_code", None)
|
|
)
|
|
flat_dict["flatisfy_time_to"] = (
|
|
flat_dict["flatisfy"].get("time_to", {})
|
|
)
|
|
flat_dict["flatisfy_constraint"] = (
|
|
flat_dict["flatisfy"].get("constraint", "default")
|
|
)
|
|
del flat_dict["flatisfy"]
|
|
|
|
flat_dict = {k: v for k, v in flat_dict.items()
|
|
if k in inspect(Flat).columns.keys()}
|
|
return Flat(**flat_dict)
|
|
|
|
def __repr__(self):
|
|
return "<Flat(id=%s, urls=%s)>" % (self.id, self.urls)
|
|
|
|
def json_api_repr(self):
|
|
"""
|
|
Return a dict representation of this flat object that is JSON
|
|
serializable.
|
|
"""
|
|
flat_repr = {
|
|
k: v
|
|
for k, v in self.__dict__.items()
|
|
if not k.startswith("_")
|
|
}
|
|
if isinstance(flat_repr["status"], FlatStatus):
|
|
flat_repr["status"] = flat_repr["status"].name
|
|
if isinstance(flat_repr["utilities"], FlatUtilities):
|
|
flat_repr["utilities"] = flat_repr["utilities"].name
|
|
|
|
return flat_repr
|