B d@sddlZddlZddlmZddlmZddlmZddlmZm Z m Z ddl m Z ddl mZd gZGd d d ejZd d ZGdd d ZdS)N) OrderedDict)datetime)Path)OptionalTextIOUnion) PegasusError)yamlWritablec@seZdZddZdS)_CustomEncodercCs8t|dr*t|jr|Std|tj||S)z if isinstance(obj, Date): return "whatever spec we come up with for Date such as ISO8601" elif isinstance(obj, Path): return obj.resolve __json__z__json__ is not callable for {})hasattrcallabler TypeErrorformatjson JSONEncoderdefault)selfobjra/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/Pegasus/api/writable.pyrs  z_CustomEncoder.defaultN)__name__ __module__ __qualname__rrrrrr sr cCs4t|tstd|tttdd|DS)a+Helper function to remove keys where their values are set to None to avoid cluttering yaml/json files :param _dict: object represented as a dict :type _dict: dict :raises ValueError: _dict must be of type dict :return: new dictionary with 'None' values removed :rtype: dict z+invalid _dict: {}; _dict must be of type {}cSs g|]\}}|dk r||fqS)Nr).0kvrrr 0sz%_filter_out_nones..) isinstancedictrrtyperitems)_dictrrr_filter_out_nones"s r%c@sZeZdZdZdddhZddZeeddd Zd d Z de e e e fe d ddZd S)r z6Derived class can be serialized to a json or yaml fileymlr rcCs d|_dS)N)_path)rrrr__init__8szWritable.__init__)returncCs&|jdkrtd|jjt|jS)z Retrieve the path to which this object has been written to. :raises PegasusError: object has not yet been written to a file :return: resolved path to which this object has been written :rtype: Path Nz@{}.write(filename) must be called before trying to retrieve path)r'r r __class__rr)rrrrpath;s  z Writable.pathcCs|tjkr"tdj|tjdtdtt dddfg}| t j t j|tdtd|d kst|d krtj||d d nt j||td dddS)a;Internal function to dump to file in either yaml or json formats :param file: file object to write to :type file: file :param _format: file format that can be "yml", "yaml", or "json :type _ext: str :raises ValueError: _format must be one of "yml", "yaml" or "json" z;invalid _ext: {_format}, extension must be one of {formats})_formatformatsz x-pegasusz%m-%d-%yT%H:%M:%SZpython)Z createdByZ createdOnZapiLang)cls)object_pairs_hookr&r T)Z allow_unicodeF)r/indent ensure_asciiN)lowerr _FORMATS ValueErrorrrgetpassgetuserrnowstrftimeupdaterloadsdumpsr r dump)rfiler,Z self_as_dictrrr_writeMs  zWritable._writeN)r?r,c Cs<|tjkrtd|dkr$|j}t|trt|}|jdd}t |d*}|tjkrl| ||n | ||WdQRXt| |_ nt |dr(y tt|j}|jdd}Wn tk r| ||YnHX|tjkr| ||n | ||t|jtr8tt|j |_ ntdj|ddS)a/Serialize this class as either yaml or json and write to the given file. If file==None, this class will be written to a default file. The following classes have these defaults: .. table:: Default Files :widths: auto ===================== =================== Class Default Filename ===================== =================== SiteCatalog sites.yml ReplicaCatalog replicas.yml TransformationCatalog transformations.yml Workflow workflow.yml ===================== =================== :param file: path or file object (opened in "w" mode) to write to, defaults to None :type file: Optional[Union[str, TextIO]] :param _format: can be either "yml", "yaml" or "json", defaults to "yml" :type _format: str, optional :raises ValueError: _format must be one of "yml", "yaml" or "json" :raises TypeError: file must be a str or file object zPinvalid file format: {_format}, format should be one of 'yml', 'yaml', or 'json'Nrwreadz){file} must be of type str or file object)r?)r4r r5r6Z_DEFAULT_FILENAMEr strrsuffixopenr@resolver'rnameAttributeErrorrr)rr?r,r+extfrrrwritezs4      zWritable.write)Nr&)rrr__doc__r5r(propertyrr+r@rrrCrrKrrrrr 3s -)r7r collectionsrrpathlibrtypingrrrerrorsr ZPegasusr __all__rr r%r rrrrs