cc_str.py

'字符串相关'

import re,sys,time,datetime,traceback
from pprint import pprint

UNITS = 'KMGTPEZYBND'

def strsize(x):
    unit = 'B'
    for du in UNITS:
        if x < 10:
            return '%.2f %s' % (x,unit)
        if x < 1000:
            return '%.1f %s' % (x,unit)
        unit = du + 'B'
        x /= 1024.0
    return '%g %s' % (x,unit)

year_len = 365.2422 #一年的天数
mon_len = year_len / 12

def strtime(a):
    a *= 1e15
    us = ['fs','ps','ns','us','ms','s','min','h', 'days','months','years','centuries']
    ks = [1000,1000,1000,1000,1000, 60,   60, 24,mon_len,      12,    100,]
    n = len(ks)
    for i in range(n):
        if a < ks[i]:
            du = us[i]
            break
        a /= ks[i]
    else:
        du = us[n]
    return '%.8g %s' % (a,du)

def strcount(n:int):
    s = str(n)
    ans = []
    while s:
        ans.append(s[-3:])
        s = s[:-3]
    ans.reverse()
    return ','.join(ans)

def log_info(s:str,f=sys.stdout):
    dt = datetime.datetime.now()
    ts = dt.isoformat(' ')
    msg = f'[{ts}]{s}\n'
    f.write(msg)
    f.flush()

def get_time_fn() -> str:
    '根据时间产生文件名'
    dt = datetime.datetime.now()
    ts = dt.isoformat('_')
    ts = re.sub(r'[/\-:]','',ts)
    ts = ts.replace('.','_')
    dt,_ = ts.split('_',1)
    return ts

class prog_logger:
    '进度提示器'
    def __init__(self):
        self.start()

    def start(self):
        self.t = time.time()

    def finish(self):
        log_info('finish')

    def log(self,i,n,delay=300,tip=''):
        dt = time.time()
        if dt < self.t + delay:return
        self.t = dt
        log_info(f'{i}/{n} {tip}')

class prog_logger2(prog_logger):
    def __init__(self):
        super().__init__()
        self.n = 0

    def log(self,dn,delay=300):
        self.n += dn
        dt = time.time()
        if dt < self.t + delay:return
        self.t = dt
        log_info(str(self.n))

    def finish(self):
        log_info(f'finish {self.n}')

def join_re_pts(pts,sep='|',add_bra=True,escape=False):
    if escape:
        pts = map(re.escape,pts)
    if add_bra:
        pts = (f'({t})' for t in pts)
    return sep.join(pts)

def show_exc():
    log_info('Exception')
    traceback.print_exc(file=sys.stdout)
    log_info('Stack')
    traceback.print_stack(file=sys.stdout)

def call_with_exc(func,*a,**k):
    try:
        func(*a,**k)
    except SystemExit:
        raise
    except:
        show_exc()
        raise

def _test():
    xs = (0,1,100,10**7,123<<30,907<<30,1008<<30,234<<180)
    for x in xs:
        log_info(strsize(x))
    for x in xs:
        log_info(strtime(x))
    print(1/0)

if __name__=='__main__':
    call_with_exc(_test)