a gz%@s0dZddlmZmZddlmZddlmZddlmZddl m Z dZ dZ dd d d d d d dddddd ddddddddddddZ eee ddZgdZeedddZeedddZeedd d!Zd.eeeeeed#fed$d%d&Zd/eeeefeeed#fed'd(d)Zd0eeed+d,d-Zd#S)1zR Spell out time as Thai words. Convert time string or time object to Thai words. )datetimetime)Union) Tokenizernum_to_thaiword)thaiword_to_numz%H:%M:%Sz%H:%M )uศูนย์uหนึ่งu สองu ยี่u สามu สี่u ห้าuหกu เจ็ดu แปดu เก้า สิบu เอ็ดโมงเช้าโมงเย็น บ่ายบ่ายโมงตีเที่ยงวันเที่ยงคืนเที่ยง ทุ่มนาฬิกาครึ่งZnewmm)Z custom_dictZengine) rrr โมงr!r rrrr)hreturncCsd}|dkr|d7}n|dkr0|dt|7}n|dkrN|t|dd7}nl|dkr`|d 7}nZ|d kr|d krz|d 7}q|d t|dd7}n&|d kr|d7}n|t|d d7}|S)zThai time (6-hour clock).rrrrrrrrrrrrr#uหกโมงเย็นr rr$textr)=/usr/local/lib/python3.9/dist-packages/pythainlp/util/time.py _format_6h>s"    r+cCsd}|dkr|d7}nx|dkr0|dt|7}n^|dkrJ|t|d7}nD|dkr\|d7}n2|d krz|t|dd7}n|t|d d 7}|S) z"Thai time (modified 6-hour clock).r&rrrrrr#rrr rr'r)r)r* _format_m6hWs  r-cCst|d}|S)zThai time (24-hour clock).r!rr'r)r)r* _format_24hks r.24hN)r$msfmt precisionr%cCsd}|dkrt|}n2|dkr(t|}n |dkr:t|}ntd||dvr|dkrz|dksh|d krz|d vrz|d 7}q|t|d 7}|d kr|t|d7}nJ|r|dkr|dkr|d vr|d 7}n|t|d 7}|r|t|d7}|S)Nr&6hm6hr/zTime format not supported: )r0r1rrr0)r4r5r"u นาทีr1uวินาที)r+r-r.NotImplementedErrorr)r$r0r1r2r3r(r)r)r*_formatqs*      r7) time_datar2r3r%c Csd}t|ttfr|}nzt|ts*td|s6tdzt|t}Wn6ty|zt|t}WntyvYn0Yn0|std|dt |j |j |j ||}|S)u Spell out time as Thai words. :param str time_data: time input, can be a datetime.time object or a datetime.datetime object or a string (in H:M or H:M:S format, using 24-hour clock) :param str fmt: time output format * *24h* - 24-hour clock (default) * *6h* - 6-hour clock * *m6h* - Modified 6-hour clock :param str precision: precision of the spell out time * *m* - always spell out at minute level * *s* - always spell out at second level * None - spell out only non-zero parts :return: Time spelled out as Thai words :rtype: str :Example: :: time_to_thaiword("8:17") # output: # แปดนาฬิกาสิบเจ็ดนาที time_to_thaiword("8:17", "6h") # output: # สองโมงเช้าสิบเจ็ดนาที time_to_thaiword("8:17", "m6h") # output: # แปดโมงสิบเจ็ดนาที time_to_thaiword("18:30", fmt="m6h") # output: # หกโมงครึ่ง time_to_thaiword(datetime.time(12, 3, 0)) # output: # สิบสองนาฬิกาสามนาที time_to_thaiword(datetime.time(12, 3, 0), precision="s") # output: # สิบสองนาฬิกาสามนาทีศูนย์วินาที NzSTime input must be a datetime.time object, a datetime.datetime object, or a string.zTime string cannot be empty.z Time string 'z%' does not match H:M or H:M:S format.) isinstancerrstr TypeError ValueErrorstrptime_TIME_FORMAT_WITH_SEC_TIME_FORMAT_WITHOUT_SECr7hourminutesecond)r8r2r3_timer(r)r)r*time_to_thaiwords,1     rDT)r(paddingr%c Cstt}|dddddd}gd}d}tD]b}||vrb|dkrb|||d}qq8||vr8|dkr8|D] }||vrv|||d}qqvq8q8d|vrtd|d}~t|d }|d } t | d krt| } nd } d}|d d kr:|d |vr:|d d r:|t t d |d d 7}n|d dkrn|d |vrn|t t|d 7}n||d dkr|d |vrt|d dkr|t t|d d7}n|t t|d 7}n|d dks|d dkr|d dkr|t t|d d7}n|d dks.|d dkrV|d |vrV|t t|d d7}n|d dkrn|d7}n||d dks|d dkr|d7}nV|d dkr|d7}n>|d dkrt |d kr|d7}n|t t|d d7}|std|rt |d krd|}|d7}| rd } | D]b}||vr*|d krP| t|7} n:|d krn| d krn| d!9} n|d kr*| d kr*| d!7} q*| d kr| d"kr|t | 7}n|dt | 7}n|d#7}|S)$u4 Convert Thai time in words into time (H:M). :param str text: Thai time in words :param bool padding: Zero pad the hour if True :return: time string :rtype: str :Example: :: thaiword_to_time"บ่ายโมงครึ่ง") # output: # 13:30 u กว่าr&uๆ )uตีหนึ่งuตีสองuตีสามuตีสี่uตีห้าr|z)Cannot find any Thai word for time affix.rr r!Nrrrr#rrr0rrZ12rZ13r Z19rz#Cannot find any Thai word for hour.:rrrZ00) list_DICT_THAI_TIMEkeysreplace_THAI_TIME_AFFIXr<split_THAI_TIME_CUTZ word_tokenizelenr:rjoin) r(rEZ keys_dictZ_irCZaffixjZ_LIST_THAI_TIMEr@rAnr)r)r*thaiword_to_times    *"**        rV)r/N)r/N)T)__doc__rrtypingrZpythainlp.tokenizerZpythainlp.util.numtowordrZpythainlp.util.wordtonumrr>r?rLrKrMrQrOintr:r+r-r.r7rDboolrVr)r)r)r*st        &   Q