从生日得到星座

#! /usr/bin/python
# -*- coding: utf-8 -*-

def get_constellation(ymd="2000-01-01"):
    stellates = [
        {'date':120, 'name':u'水瓶座'},
        {'date':219, 'name':u'双鱼座'},
        {'date':321, 'name':u'牡羊座'},
        {'date':420, 'name':u'金牛座'},
        {'date':521, 'name':u'双子座'},
        {'date':622, 'name':u'巨蟹座'},
        {'date':723, 'name':u'狮子座'},
        {'date':823, 'name':u'处女座'},
        {'date':923, 'name':u'天秤座'},
        {'date':1024, 'name':u'天蝎座'},
        {'date':1123, 'name':u'射手座'},
        {'date':1222, 'name':u'魔羯座'}
    ]
    if not ymd or ymd == "0000-00-00":
        return u'未知'
    ymd = int(ymd[5:7]+ymd[8:10])
    index = ymd / 100 - 1
    if ymd >= stellates[index]['date']:
        return stellates[index]['name']
    else:
        return stellates[index-1]['name']