def fake_object(self, data):
"""Create a fake instance of Addon and related models from ES data."""
obj = Addon(id=data["id"], slug=data["slug"], is_listed=True)
# Attach base attributes that have the same name/format in ES and in
# the model.
self._attach_fields(
obj,
data,
(
"average_daily_users",
"bayesian_rating",
"created",
"default_locale",
"guid",
"hotness",
"icon_type",
"is_listed",
"last_updated",
"modified",
"public_stats",
"slug",
"status",
"type",
"weekly_downloads",
),
)
# Temporary hack to make sure all add-ons have a modified date when
# serializing, to avoid errors when calling get_icon_url().
# Remove once all add-ons have been reindexed at least once since the
# addition of `modified` in the mapping.
if obj.modified is None:
obj.modified = obj.created
# Attach attributes that do not have the same name/format in ES.
obj.tag_list = data["tags"]
obj.disabled_by_user = data["is_disabled"] # Not accurate, but enough.
# Categories are annoying, skip them for now. We probably need to start
# declaring them in the code to properly handle translations etc if we
# want to display them in search results.
obj.all_categories = []
# Attach translations (they require special treatment).
self._attach_translations(obj, data, self.translated_fields)
# Attach related models (also faking them).
data_version = data.get("current_version")
if data_version:
obj._current_version = Version(
addon=obj,
id=data_version["id"],
reviewed=self.handle_date(data_version["reviewed"]),
version=data_version["version"],
)
data_files = data_version.get("files", [])
obj._current_version.all_files = [
File(
id=file_["id"],
created=self.handle_date(file_["created"]),
hash=file_["hash"],
filename=file_["filename"],
platform=file_["platform"],
size=file_["size"],
status=file_["status"],
version=obj._current_version,
)
for file_ in data_files
]
# In ES we store integers for the appversion info, we need to
# convert it back to strings.
compatible_apps = {}
for app_id, compat_dict in data["appversion"].items():
app_name = APPS_ALL[int(app_id)]
compatible_apps[app_name] = ApplicationsVersions(
min=AppVersion(version=compat_dict.get("min_human", "")),
max=AppVersion(version=compat_dict.get("max_human", "")),
)
obj._current_version.compatible_apps = compatible_apps
if data["type"] == amo.ADDON_PERSONA:
persona_data = data.get("persona")
if persona_data:
obj.persona = Persona(
addon=obj,
accentcolor=persona_data["accentcolor"],
display_username=persona_data["author"],
header=persona_data["header"],
footer=persona_data["footer"],
persona_id=1 if persona_data["is_new"] else None,
textcolor=persona_data["textcolor"],
)
else:
# Sadly, https://code.djangoproject.com/ticket/14368 prevents
# us from setting obj.persona = None. This is fixed in
# Django 1.9, but in the meantime, work around it by creating
# a Persona instance with a custom attribute indicating that
# it should not be used.
obj.persona = Persona()
obj.persona._broken = True
return obj
def fake_object(self, data):
"""Create a fake instance of Addon and related models from ES data."""
obj = Addon(id=data['id'], slug=data['slug'], is_listed=True)
# Attach base attributes that have the same name/format in ES and in
# the model.
self._attach_fields(
obj, data,
('average_daily_users', 'bayesian_rating', 'created',
'default_locale', 'guid', 'hotness', 'icon_type', 'is_listed',
'last_updated', 'modified', 'public_stats', 'slug', 'status',
'type', 'weekly_downloads'))
# Temporary hack to make sure all add-ons have a modified date when
# serializing, to avoid errors when calling get_icon_url().
# Remove once all add-ons have been reindexed at least once since the
# addition of `modified` in the mapping.
if obj.modified is None:
obj.modified = obj.created
# Attach attributes that do not have the same name/format in ES.
obj.tag_list = data['tags']
obj.disabled_by_user = data['is_disabled'] # Not accurate, but enough.
# Categories are annoying, skip them for now. We probably need to start
# declaring them in the code to properly handle translations etc if we
# want to display them in search results.
obj.all_categories = []
# Attach translations (they require special treatment).
self._attach_translations(obj, data, self.translated_fields)
# Attach related models (also faking them).
data_version = data.get('current_version')
if data_version:
obj._current_version = Version(addon=obj,
id=data_version['id'],
reviewed=self.handle_date(
data_version['reviewed']),
version=data_version['version'])
data_files = data_version.get('files', [])
obj._current_version.all_files = [
File(id=file_['id'],
created=self.handle_date(file_['created']),
hash=file_['hash'],
filename=file_['filename'],
platform=file_['platform'],
size=file_['size'],
status=file_['status'],
version=obj._current_version) for file_ in data_files
]
# In ES we store integers for the appversion info, we need to
# convert it back to strings.
compatible_apps = {}
for app_id, compat_dict in data['appversion'].items():
app_name = APPS_ALL[int(app_id)]
compatible_apps[app_name] = ApplicationsVersions(
min=AppVersion(version=compat_dict.get('min_human', '')),
max=AppVersion(version=compat_dict.get('max_human', '')))
obj._current_version.compatible_apps = compatible_apps
if data['type'] == amo.ADDON_PERSONA:
persona_data = data.get('persona')
if persona_data:
obj.persona = Persona(
addon=obj,
accentcolor=persona_data['accentcolor'],
display_username=persona_data['author'],
header=persona_data['header'],
footer=persona_data['footer'],
persona_id=1 if persona_data['is_new'] else None,
textcolor=persona_data['textcolor'])
else:
# Sadly, https://code.djangoproject.com/ticket/14368 prevents
# us from setting obj.persona = None. This is fixed in
# Django 1.9, but in the meantime, work around it by creating
# a Persona instance with a custom attribute indicating that
# it should not be used.
obj.persona = Persona()
obj.persona._broken = True
return obj
