Graphviz自动根据文件生成时间序列图脚本和示例demo文件

#!/usr/bin/python # -*- coding: utf-8 -*- import sys msi=int(sys.argv[1]) dsi=int(sys.argv[2]) mei=int(sys.argv[3]) dei=int(sys.argv[4]) file_path=sys.argv[5] year="23" mdays=[31,28,31, 30,31,30, 31,31,30, 31,30,31] def gen_mon(msi,mei,dsi,dei): print("\nnode [fontsize=20, shape=box, style=filled fillcolor=lightgreen];") for m in range(msi, mei+1): if m == msi: ds=int(dsi) else: ds=1 if m==mei: de=dei if de > mdays[m-1]: de = mdays[m-1] else: de=mdays[m-1] print("\n{:>02}.{:>02}".format(year,m)) def gen_day(msi,mei,dsi,dei): print("\nnode [fontsize=20, shape=box, style=filled fillcolor=lightgray];") for m in range(msi, mei+1): if m == msi: ds=int(dsi) else: ds=1 if m==mei: de=dei if de > mdays[m-1]: de = mdays[m-1] else: de=mdays[m-1] print("\n{:>02}.{:>02}->".format(year,m)) for i in range(ds, de+1): if i!=de: print("{:>02}{:>02}->".format(m,i)) else: print("{:>02}{:>02}".format(m,i)) def gen_rank_same(msi,mei,dsi,dei): print("# rank") print("\nnode [fontsize=20, shape=box, style=filled fillcolor=cyan];") for m in range(msi, mei+1): if m == msi: ds=int(dsi) else: ds=1 if m==mei: de=dei if de > mdays[m-1]: de = mdays[m-1] else: de=mdays[m-1] print("\n#{:>02}.{:>02}".format(year,m)) septimelist=[3,6,9,12] seasonlist=[3,6,9,12] for i in range(ds, de+1): if (m in seasonlist) and (i == 30): print("{ rank=same; %02d%02d -> season_%02d%02d}" %(m,i, m,i)) elif (m in septimelist) and (i==22): print("{ rank=same; %02d%02d -> jieqi_zhifen%02d%02d}" %(m,i, m,i)) else: print("{ rank=same; %02d%02d }" %(m,i)) def gen_head(): print("digraph timeline {") print('size = "7,8";') def gen_tail(): print("}") def _print_riji_event_label(lines): # 遍历每一行,打印label print("\n# event label") for line in lines: if len(line) < 10: continue # 2023-09-12 周一；Graphviz年时间线 def ldate=line[5:7]+line[8:10] print("\n## " + ldate) line=line[10:] line = line.replace('\n','').replace("；", ";").replace(",",";") #逗号也是一个分隔符 la=line.split(';') for i in range(0, len(la)): if len(la[i]) < 2: continue if i == 0: #第一个不标数字 是星期 seq="" weekenddaylist=["周六", "周日"] la[i]=la[i].replace(" ",'').replace(",", '') if la[i] in weekenddaylist: weekendcolor="style=filled fillcolor=lightgreen" else: weekendcolor="style=filled fillcolor=coral" else: seq="{:>02d}:".format(i) weekendcolor="" print("d_env_"+ldate + "_" + str(i) + " [label=\"" + seq + la[i] + "\" " + weekendcolor +"];") def _print_riji_event_ranksame(lines): print("\n# ranks=same") for line in lines: if len(line) < 10: continue ldate=line[5:7]+line[8:10] line=line[9:] line = line.replace('\n','').replace("；", ";").replace(",",";") #逗号也是一个分隔符 la=line.split(';') print("{ rank=same; " + ldate), for i in range(0, len(la)): if len(la[i]) < 2: continue print("-> d_env_" + ldate + "_" + str(i)), print("}") def gen_riji_event(): print("\nnode [fontsize=20, shape=box, style=filled fillcolor=lightblue];") #读取文件riji.log到list中 #file_path = "riji.log" with open(file_path, "r") as file: lines=file.readlines() #linearray=lines.split('\n') _print_riji_event_label(lines) _print_riji_event_ranksame(lines) def gen_total(): gen_head() gen_mon(msi, mei, dsi, dei) gen_day(msi, mei, dsi, dei) gen_rank_same(msi, mei, dsi, dei) gen_riji_event() gen_tail() def main(): gen_total() main() #python riji.py 9 1 9 31 > 1.gv