<s> from enum import Enum <EOL> class AtomState ( str , Enum ) : <EOL> NOT_STARTED = '<STR_LIT>' <EOL> IN_PROGRESS = '<STR_LIT>' <EOL> COMPLETED = '<STR_LIT>' <EOL> class Atom ( object ) : <EOL> def __init__ ( <EOL> self , <EOL> command_string , <EOL> expected_time = None , <EOL> actual_time = None , <EOL> exit_code = None , <EOL> state = None , <EOL> atom_id = None , <EOL> subjob_id = None <EOL> ) : <EOL> """<STR_LIT>""" <EOL> self . command_string = command_string <EOL> self . expected_time = expected_time <EOL> self . actual_time = actual_time <EOL> self . exit_code = exit_code <EOL> self . state = state <EOL> self . subjob_id = subjob_id <EOL> self . id = atom_id <EOL> def api_representation ( self ) : <EOL> return { <EOL> '<STR_LIT>' : self . command_string , <EOL> '<STR_LIT>' : self . expected_time , <EOL> '<STR_LIT>' : self . actual_time , <EOL> '<STR_LIT>' : self . exit_code , <EOL> '<STR_LIT:state>' : self . state , <EOL> '<STR_LIT:id>' : self . id , <EOL> '<STR_LIT>' : self . subjob_id , <EOL> } </s>
