#!/usr/bin/env python
import logging, os, sys
from theano import config
from theano.gof.cc import get_module_cache

_logger = logging.getLogger('theano.bin.theano-cache')
_logger.setLevel(logging.WARN)


if len(sys.argv) == 1:
    print config.compiledir
elif sys.argv[1] in ('clear'):
    # We skip the refresh on module cache creation because the refresh will
    # be done when calling clear afterwards.
    cache = get_module_cache(init_args=dict(do_refresh=False))
    cache.clear(unversioned_min_age=-1, clear_base_files=True,
                delete_if_problem=True)
    # Print a warning if some cached modules were not removed, so that the user
    # knows he should manually delete them to properly clear the cache.
    items = [item for item in sorted(os.listdir(cache.dirname))
                  if item.startswith('tmp')]
    if items:
        _logger.warning('There remain elements in the cache dir that you may '
                        'need to erase manually. The cache dir is:\n  %s' %
                        config.compiledir)
        _logger.debug('Remaining elements (%s): %s' %
                      (len(items), ', '.join(items)))

else:
    print 'command "%s" not recognized' % sys.argv[1]
    print 'Type "theano-cache" to print the cache location'
    print 'Type "theano-cache clear" to erase the cache'
    sys.exit(1)

