DouKaku?

与えられた文字列から特定の条件を満たす文字列を抽出するコードを書いてください。 状況としてはテキスト形式で渡された原稿の中から、画像のファイル名を抽出するようなものをイメージしてください。

サンプル入力

aaa abc-hidden.png>hoge-big.jpeg
---foo-hidden-small.gif|^_^a.bmp
--hiddena-hoge.png<=not hidden~~
--small.jpg<=not small(^_^)
normal-small-big.hoge

サンプル出力

name:'abc', ext:'png', size: normal hidden: True
name:'hoge', ext:'jpeg', size: big hidden: False
name:'foo', ext:'gif', size: small hidden: True
name:'a', ext:'bmp', size: normal hidden: False
name:'hoge', ext:'png', size: normal hidden: False
name:'small', ext:'jpg', size: normal hidden: False
name:'small', ext:'hoge', size: big hidden: False

探すべき文字列は下の条件を満たします

• アルファベットと1個のピリオド、ハイフンで構成される
• 前後にはアルファベットではない文字がある(abcd.jpgがaaaabcd.jpghogeなどと書かれていることはない)
• ピリオドの後ろは拡張子で、アルファベットだけで構成されている
• ピリオドの直前に-hidden, -small, -bigがある場合には特殊な意味がある。複数個ある場合(a-hidden-big.jpgなど)も同じ
• ファイル名に-hiddenと-smallまたは-hiddenと-bigの両方が含まれる場合は-hiddenの方が先にある
• 特殊な意味の-hidden, -small, -big以外でハイフンが使われることはない
• 特殊な意味の-smallと-bigの両方が付くことはない

出力は以下の条件を満たす必要があります

• ファイル名が出現した順に表示される
• ファイル名に-hiddenが含まれるかどうかを真偽値で表示する
• ファイル名に-smallまたは-bigが含まれる場合はsmallまたはbigと、含まれない場合はnormalと表示する
• -hidden, -small, -bigを取り除いたファイル名部分と、拡張子を表示する

このお題は、正規表現のグループに名前をつけて連想配列として取得できるPythonからの挑戦状です。

すみません、名前を付けた正規表現を知らないので、とりあえず普通の正規表現で。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17

import re

def extract(s):
    for m in re.finditer('([a-zA-Z]+)(-hidden)?(?:-(big|small))?\.([a-zA-Z]+)', s):
        print "name:%r, ext:%r, size: %s hidden: %r" % \
            (m.group(1), m.group(4), m.group(3) or "normal", bool(m.group(2)))

def main():
    extract("""\
aaa abc-hidden.png>hoge-big.jpeg
---foo-hidden-small.gif|^_^a.bmp
--hiddena-hoge.png<=not hidden~~
--small.jpg<=not small(^_^)
normal-small-big.hoge""")

if __name__ == '__main__':
    main()

名前つき正規表現なら、大体こんな感じになると思いますが、
このお題の「-hidden」の処理のように、なんらかの操作が
必要な場合は、ありがたみが少々落ちますね。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13

import re

dat = '''aaa abc-hidden.png>hoge-big.jpeg
---foo-hidden-small.gif|^_^a.bmp
--hiddena-hoge.png<=not hidden~~
--small.jpg<=not small(^_^)
normal-small-big.hoge
'''
ptn = '(?P<name>[a-zA-Z]+)(-hidden)?(?:-(?P<size>big|small))?\.(?P<ext>[a-zA-Z]+)'
fmt = "name:'%(name)s', ext:'%(ext)s', size: %(size)s, hidden: %%s"

for g in re.finditer(ptn, dat):
  print fmt % g.groupdict('normal') % bool(g.group(2))

それは%の文字列フォーマット操作で全部やってしまおうとせずに
素直にif文を使う方が後々読みやすくていいと思います。

1
2
3
4
5
6
7
8

ptn = '(?P<name>[a-zA-Z]+)(?P<hidden>-hidden)?(?:-(?P<size>big|small))?\.(?P<ext>[a-zA-Z]+)'
fmt = "name:'%(name)s', ext:'%(ext)s', size: %(size)s, hidden: %(hidden)s"

for g in re.finditer(ptn, dat):
	params = g.groupdict()
	params["hidden"] = (params["hidden"] == "-hidden")
	if not params["size"]: params["size"] = "normal"
	print fmt % params