Accords-Library-Archive
/
gallery.accords-library.com
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

server/posts: implement FLAG_MAP

This commit is contained in:
rr- 2016-05-10 11:57:05 +02:00
parent 4ec842024c
commit ba3678dec1
2 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
server/szurubooru/db/post.py
View File

@ -91,7 +91,6 @@ class Post(Base):
TYPE_FLASH = 'flash' TYPE_FLASH = 'flash'
FLAG_LOOP = 'loop' FLAG_LOOP = 'loop'
ALL_FLAGS = [FLAG_LOOP]
# basic meta # basic meta
post_id = Column('id', Integer, primary_key=True) post_id = Column('id', Integer, primary_key=True)

15
server/szurubooru/func/posts.py
View File

@ -30,7 +30,9 @@ TYPE_MAP = {
db.Post.TYPE_VIDEO: 'video', db.Post.TYPE_VIDEO: 'video',
db.Post.TYPE_FLASH: 'flash', db.Post.TYPE_FLASH: 'flash',
} }
# TODO: FLAG_MAP FLAG_MAP = {
db.Post.FLAG_LOOP: 'loop',
}
def get_post_content_url(post): def get_post_content_url(post):
return '%s/posts/%d.%s' % ( return '%s/posts/%d.%s' % (
@ -150,7 +152,7 @@ def update_post_safety(post, safety):
safety = util.flip(SAFETY_MAP).get(safety, None) safety = util.flip(SAFETY_MAP).get(safety, None)
if not safety: if not safety:
raise InvalidPostSafetyError( raise InvalidPostSafetyError(
'Safety can be either of %r.', list(SAFETY_MAP.values())) 'Safety can be either of %r.' % list(SAFETY_MAP.values()))
post.safety = safety post.safety = safety
def update_post_source(post, source): def update_post_source(post, source):
@ -261,11 +263,14 @@ def update_post_notes(post, notes):
db.PostNote(polygon=note['polygon'], text=note['text'])) db.PostNote(polygon=note['polygon'], text=note['text']))
def update_post_flags(post, flags): def update_post_flags(post, flags):
target_flags = []
for flag in flags: for flag in flags:
if flag not in db.Post.ALL_FLAGS: flag = util.flip(FLAG_MAP).get(flag, None)
if not flag:
raise InvalidPostFlagError( raise InvalidPostFlagError(
'Flag must be one of %r.' % db.Post.ALL_FLAGS) 'Flag must be one of %r.' % list(FLAG_MAP.values()))
post.flags = flags target_flags.append(flag)
post.flags = target_flags
def feature_post(post, user): def feature_post(post, user):
post_feature = db.PostFeature() post_feature = db.PostFeature()