https://github.com/dereneaton/ipyrad
Tip revision: a1c0cfc25956e7615f9d9c773deeab38db08a2a0 authored by deren on 15 December 2016, 16:53:46 UTC
"Updating ipyrad/__init__.py to version - 0.5.13
"Updating ipyrad/__init__.py to version - 0.5.13
Tip revision: a1c0cfc
__init__.py
#!/usr/bin/env ipython2
# pylint: disable=C0103
import os as _os
import atexit
## define state vars
__interactive__ = 1 ## CLI __main__ changes to 0
__version__ = "0.5.13"
## Possible values for __loglevel__: "DEBUG" "INFO" "WARN" "ERROR"
__debugflag__ = "./.debug"
__debugfile__ = "./ipyrad_log.txt"
## debug is set based on whether the flag exists
if _os.path.exists(__debugflag__):
__loglevel__ = "DEBUG"
else:
__loglevel__ = "ERROR"#"INFO"
## main ip.functions
from . import load
from . import assemble
from . import file_conversion
from .load import save_json
from .load import load_json
#from . import plotting ## do not autoimport plotting, import as ipp
#from . import analysis ## do not autoimport analysis, import as ipa
## bring nested functions to ip.
from ipyrad.core.assembly import Assembly
from ipyrad.core.assembly import merge
from ipyrad.core.sample import Sample
from ipyrad.core.paramsinfo import paramsinfo
####################################################################
## create logger for debugging
## this needs to come after __loglevel__ definition
## sets log config and prints warning if __loglevel__ is in hackers mode
import logging as _logging
import logging.config as _lconfig
## clear the logfile if it is too big
if _os.path.exists(__debugfile__):
if _os.path.getsize(__debugfile__) > 5000000:
with open(__debugfile__, 'w') as clear:
clear.write("file reset")
## check that all dependencies exist and are working
import subprocess as _subprocess
import sys as _sys
_LOGGER = _logging.getLogger(__name__)
if __loglevel__ == "DEBUG":
_LOGGER.debug("Engine init")
def debug_on():
"""
Turns on debugging by creating hidden tmp file
This is only run by the __main__ engine.
"""
## make tmp file and set loglevel for top-level init
with open(__debugflag__, 'w') as dfile:
dfile.write("wat")
__loglevel__ = "DEBUG"
_LOGGER.info("debugging turned on and registered to be turned off at exit")
_set_debug_dict(__loglevel__)
def _set_debug_dict(__loglevel__):
""" set the debug dict """
_lconfig.dictConfig({
'version': 1,
'disable_existing_loggers': False,
'formatters': {
'standard': {
'format': "%(asctime)s \t"\
+"pid=%(process)d \t"\
+"[%(filename)s]\t"\
+"%(levelname)s \t"\
+"%(message)s"
},
},
'handlers': {
__name__: {
'level':__loglevel__,
'class':'logging.FileHandler',
'filename':__debugfile__,
'formatter':"standard",
'mode':'a+'
}
},
'loggers':{
__name__: {
'handlers': [__name__],
'level': __loglevel__,
'propogate': True
}
}
})
_set_debug_dict(__loglevel__)
def debug_off():
""" turns off debugging by removing hidden tmp file """
if _os.path.exists(__debugflag__):
_os.remove(__debugflag__)
def _cmd_exists(cmd):
""" check if dependency program is there """
return _subprocess.call("type " + cmd,
shell=True,
stdout=_subprocess.PIPE,
stderr=_subprocess.PIPE) == 0
def _getbins():
""" gets the right version of vsearch, muscle, and smalt
depending on linux vs osx """
# Return error if system is 32-bit arch.
# This is straight from the python docs:
# https://docs.python.org/2/library/platform.html#cross-platform
if not _sys.maxsize > 2**32:
_sys.exit("ipyrad requires 64bit architecture")
## get platform mac or linux
_platform = _sys.platform
## get current location
path = _os.path.abspath(_os.path.dirname(__file__))
## find bin directory
ipyrad_path = _os.path.dirname(path)
bin_path = _os.path.join(ipyrad_path, "bin")
## get the correct binaries
if 'linux' in _platform:
vsearch = _os.path.join(
_os.path.abspath(bin_path),
"vsearch-linux-x86_64")
muscle = _os.path.join(
_os.path.abspath(bin_path),
"muscle-linux-x86_64")
smalt = _os.path.join(
_os.path.abspath(bin_path),
"smalt-linux-x86_64")
bwa = _os.path.join(
_os.path.abspath(bin_path),
"bwa-linux-x86_64")
samtools = _os.path.join(
_os.path.abspath(bin_path),
"samtools-linux-x86_64")
bedtools = _os.path.join(
_os.path.abspath(bin_path),
"bedtools-linux-x86_64")
qmc = _os.path.join(
_os.path.abspath(bin_path),
"wQMC-linux-x86_64")
else:
vsearch = _os.path.join(
_os.path.abspath(bin_path),
"vsearch-osx-x86_64")
muscle = _os.path.join(
_os.path.abspath(bin_path),
"muscle-osx-x86_64")
smalt = _os.path.join(
_os.path.abspath(bin_path),
"smalt-osx-x86_64")
bwa = _os.path.join(
_os.path.abspath(bin_path),
"bwa-osx-x86_64")
samtools = _os.path.join(
_os.path.abspath(bin_path),
"samtools-osx-x86_64")
bedtools = _os.path.join(
_os.path.abspath(bin_path),
"bedtools-osx-x86_64")
## only one compiled version available, works for all?
qmc = _os.path.join(
_os.path.abspath(bin_path),
"wQMC-linux-x86_64")
# Test for existence of binaries
assert _cmd_exists(muscle), "muscle not found here: "+muscle
assert _cmd_exists(vsearch), "vsearch not found here: "+vsearch
assert _cmd_exists(smalt), "smalt not found here: "+smalt
assert _cmd_exists(bwa), "bwa not found here: "+bwa
assert _cmd_exists(samtools), "samtools not found here: "+samtools
assert _cmd_exists(bedtools), "bedtools not found here: "+bedtools
#assert _cmd_exists(qmc), "wQMC not found here: "+qmc
return vsearch, muscle, smalt, bwa, samtools, bedtools, qmc
## create globals for binaries that can be accessed as: ipyrad.bins.muscle
bins = assemble.util.ObjDict()
binnames = ["vsearch", "muscle", "smalt", "bwa", "samtools", "bedtools", "qmc"]
for binn, binx in zip(binnames, _getbins()):
bins[binn] = binx
## clean up for the API
del binnames, binn, binx
if __name__ == "__main__":
pass