add db python code