diff --git a/fanficfare/adapters/adapter_masseffect2in.py b/fanficfare/adapters/adapter_masseffect2in.py index 8eb97bd..b8c924a 100644 --- a/fanficfare/adapters/adapter_masseffect2in.py +++ b/fanficfare/adapters/adapter_masseffect2in.py @@ -197,6 +197,9 @@ class MassEffect2InAdapter(BaseSiteAdapter): # If any chapter is adult, consider the whole story adult. if chapter.isAdult(): self.story.setMetadata('is_adult', True) + warning = chapter.getWarning() + if warning: + self.story.extendList('warnings', [warning]) chapterTitle = re.sub(garbagePattern, u'', chapter.getHeading()[chapterTitleStart:]) @@ -346,6 +349,11 @@ class Chapter(object): return True return False + def getWarning(self): + attributes = self.__getAttributes() + if 'warning' in attributes: + return attributes['warning'] + def getCharacters(self): return self._getListAttribute('characters')