#!/usr/bin/env python
# coding: utf-8
from ase.io import read
from matador.export import doc2res
from matador.utils.print_utils import print_notify, print_success, print_failure
from matador.utils.ase_utils import ase2dict
from sys import argv

fnames = argv[1:]
for fname in fnames:
    print_notify('Reading ' + fname)
    try:
        atoms = read(fname)
        doc = ase2dict(atoms)
        success = True
    except Exception as exc:
        success = False
    if success:
        doc2res(doc, fname.replace('.cif', '.res'), spoof_titl=True)
        print_success('Wrote .res file to ' + fname.replace('.cif', '') + '.res')
    else:
        print(doc)
        print_failure('Unable to read structure in ' + fname + '.')

print_success('Completed!')
