arvi 0.1.8__py3-none-any.whl → 0.1.11__py3-none-any.whl

arvi/utils.py

@@ -8,12 +8,20 @@ except ImportError:
     except ImportError as e:
         raise e
 
+import subprocess
 import logging
 from glob import glob
-from numpy import array
+import numpy as np
+
+try:
+    from tqdm import tqdm, trange
+except ImportError:
+    tqdm = lambda x, *args, **kwargs: x
+    trange = lambda *args, **kwargs: range(*args, **kwargs)
 
 
 def create_directory(directory):
+    """ Create a directory if it does not exist """
     if not os.path.isdir(directory):
         os.makedirs(directory)
 
@@ -53,6 +61,30 @@ def all_logging_disabled():
     finally:
         logging.disable(previous_level)
 
+def strtobool(val):
+    """Convert a string representation of truth to true (1) or false (0).
+
+    True values are 'y', 'yes', 't', 'true', 'on', and '1'; false values
+    are 'n', 'no', 'f', 'false', 'off', and '0'.  Raises ValueError if
+    'val' is anything else.
+    """
+    val = val.lower()
+    if val in ('y', 'yes', 't', 'true', 'on', '1'):
+        return 1
+    elif val in ('n', 'no', 'f', 'false', 'off', '0'):
+        return 0
+    else:
+        raise ValueError("invalid truth value {!r}".format(val))
+
+def there_is_internet(timeout=1):
+    from socket import create_connection
+    try:
+        create_connection(('8.8.8.8', 53), timeout=timeout)
+        return True
+    except OSError:
+        pass
+    return False
+
 def find_data_file(file):
     here = os.path.dirname(os.path.abspath(__file__))
     data_file = os.path.join(here, '..', 'data', file)
@@ -73,10 +105,28 @@ def ESPRESSO_ADC_issues():
     adc_file = find_data_file('obs_affected_ADC_issues.dat')
     lines = [line.strip() for line in open(adc_file).readlines()]
     file_roots = [line.split()[1] for line in lines if not line.startswith('#')]
-    return array(file_roots)
+    return np.array(file_roots)
 
 def ESPRESSO_cryostat_issues():
     cryostat_file = find_data_file('obs_affected_blue_cryostat_issues.dat')
     lines = [line.strip() for line in open(cryostat_file).readlines()]
     file_roots = [line.split()[1] for line in lines if not line.startswith('#')]
-    return array(file_roots)
+    return np.array(file_roots)
+
+
+def get_max_berv_span(self, n=None):
+    """
+    Return the indices of the n observations which maximize the BERV span.
+    If n is None, return all indices sorted by BERV span.    
+    """
+    berv_argsort = np.argsort(self.berv)
+
+    # if n is None:
+    #     n = self.N // 2
+    inds = []
+    for b1, b2 in zip(berv_argsort[:self.N // 2], berv_argsort[::-1]):
+        inds.append(b1)
+        inds.append(b2)
+    return np.array(inds[:n])
+
+

{arvi-0.1.8.dist-info → arvi-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arvi
-Version: 0.1.8
+Version: 0.1.11
 Summary: The Automated RV Inspector
 Author-email: João Faria <joao.faria@unige.ch>
 License: MIT

arvi-0.1.11.dist-info/RECORD

@@ -0,0 +1,29 @@
+arvi/HZ.py,sha256=btqDFTxHeQlmvXOQKrjOpP8YFtmFjGhSZwsYYehLOaM,2885
+arvi/__init__.py,sha256=XjBsNGlwuLftPK50u7SOmYKpmOpt9X5AIqA9LdZ8GyU,344
+arvi/binning.py,sha256=G1nZvH2ev02quDPZCReygLkH7tX6B3hFAOoOYTfhLec,15249
+arvi/config.py,sha256=XqLmIsPl9dJvco1ngANG5W868BexfuJJEaRkuKZsAK8,43
+arvi/dace_wrapper.py,sha256=bfNzHKQUryCkZsJdEJImJY3oLIwVjflJOFQRRJ4RoAE,10125
+arvi/extra_data.py,sha256=WEEaYeLh52Zdv0uyHO72Ys5MWS3naTAP4wJV2BJ1mbk,2551
+arvi/instrument_specific.py,sha256=gDNu6xM0nL3AItAxipTdg3dIGaGGi7tNWE5nnad5MxM,2873
+arvi/lbl_wrapper.py,sha256=_ViGVkpakvuBR_xhu9XJRV5EKHpj5Go6jBZGJZMIS2Y,11850
+arvi/nasaexo_wrapper.py,sha256=xrdF4KKXmlZAqu4IQtdcqfOeUXJXoui5M-eZpnVXNlk,7311
+arvi/plots.py,sha256=J1FR-7NgWL0zdjtD0Sumj6ZHLXSRC_EtnpXkTguxQ-s,17549
+arvi/programs.py,sha256=OHmOW1BJUUg7xmzpTDzyAunxiGZ77QpVIN4Z9HxmE4M,3556
+arvi/reports.py,sha256=yrdajC-zz5_kH1Ucc6cU70DK-5dpG0Xyeru-EITKpNo,3355
+arvi/setup_logger.py,sha256=pBzaRTn0hntozjbaRVx0JIbWGuENkvYUApa6uB-FsRo,279
+arvi/simbad_wrapper.py,sha256=F_VOHBI-KqT9q06sO2n2F1Ov6Un_1gAMP3ItBMqx-Yw,5284
+arvi/stats.py,sha256=MQiyLvdiAFxIC29uajTy8kuxD-b2Y6mraL4AfWkRJkM,2576
+arvi/timeseries.py,sha256=IvpjJqRL97bOwm4RpAGqvevkySj0Odjzx6vPtvlKzSI,52294
+arvi/translations.py,sha256=eyUJei8wlsyBecTz8E_ntpP35-Mre2Tzm_mUMMGaZWY,150
+arvi/utils.py,sha256=ay3GouZYqn66zF43KgSs8Ngsfb4YInKZEmk70YqVDoA,3676
+arvi/data/info.svg,sha256=0IMI6W-eFoTD8acnury79WJJakpBwLa4qKS4JWpsXiI,489
+arvi/data/obs_affected_ADC_issues.dat,sha256=pbwzvRBjS8KsLrKi1Cdgfihvo3hK3FUN8Csdysw-vAw,10653
+arvi/data/obs_affected_blue_cryostat_issues.dat,sha256=z4AK17xfz8tGTDv1FjRvQFnio4XA6PNNfDXuicewHk4,1771
+arvi/data/extra/HD86226_PFS1.rdb,sha256=vfAozbrKHM_j8dYkCBJsuHyD01KEM1asghe2KInwVao,3475
+arvi/data/extra/HD86226_PFS2.rdb,sha256=F2P7dB6gVyzCglUjNheB0hIHVClC5RmARrGwbrY1cfo,4114
+arvi/data/extra/metadata.json,sha256=C69hIw6CohyES6BI9vDWjxwSz7N4VOYX0PCgjXtYFmU,178
+arvi-0.1.11.dist-info/LICENSE,sha256=6JfQgl7SpM55t0EHMFNMnNh-AdkpGW25MwMiTnhdWQg,1068
+arvi-0.1.11.dist-info/METADATA,sha256=cb2feqRY6mVlakZeoO86IYmPQzFLTWP8IboRNaBKVrw,1306
+arvi-0.1.11.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+arvi-0.1.11.dist-info/top_level.txt,sha256=4EeiKDVLD45ztuflTGfQ3TU8GVjJg5Y95xS5XjI-utU,5
+arvi-0.1.11.dist-info/RECORD,,

{arvi-0.1.8.dist-info → arvi-0.1.11.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.42.0)
+Generator: bdist_wheel (0.43.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

arvi-0.1.8.dist-info/RECORD

@@ -1,24 +0,0 @@
-arvi/__init__.py,sha256=ArHfJ-BkFZeYWW20EaBUAzsxanci3-sI1M12IvtIVa4,223
-arvi/binning.py,sha256=rGTisXmVRzo-7PxLKOEF1DxHBVH8X-kZTYqbN_2IXmw,14693
-arvi/config.py,sha256=VM23kaAuyOWp9clfCIztJEnT5BWbG7o8see0UEqDeK0,20
-arvi/dace_wrapper.py,sha256=Db6qnrIQbf-x8mbJFYfRNHzjAaJmvYSSC-j9IyNAsts,9390
-arvi/instrument_specific.py,sha256=gDNu6xM0nL3AItAxipTdg3dIGaGGi7tNWE5nnad5MxM,2873
-arvi/lbl_wrapper.py,sha256=_ViGVkpakvuBR_xhu9XJRV5EKHpj5Go6jBZGJZMIS2Y,11850
-arvi/nasaexo_wrapper.py,sha256=xrdF4KKXmlZAqu4IQtdcqfOeUXJXoui5M-eZpnVXNlk,7311
-arvi/plots.py,sha256=9tgleBrpNhQqRTmRx_sZTqSm2DKO08NZzMPhj9rh4WY,11405
-arvi/programs.py,sha256=OHmOW1BJUUg7xmzpTDzyAunxiGZ77QpVIN4Z9HxmE4M,3556
-arvi/reports.py,sha256=yrdajC-zz5_kH1Ucc6cU70DK-5dpG0Xyeru-EITKpNo,3355
-arvi/setup_logger.py,sha256=pBzaRTn0hntozjbaRVx0JIbWGuENkvYUApa6uB-FsRo,279
-arvi/simbad_wrapper.py,sha256=OEJOZOjIBLkEbJhJK91h18i2WswfMnYqR15A-wPAbmQ,4675
-arvi/stats.py,sha256=qcTrfDXpw5tvPQci9r4ODkI5aSC-U0sOHX55i4Ipww0,1515
-arvi/timeseries.py,sha256=it4WJJypIvi9RbGcOGGykb02Z-217xaXzYwFh9153tQ,41213
-arvi/translations.py,sha256=eyUJei8wlsyBecTz8E_ntpP35-Mre2Tzm_mUMMGaZWY,150
-arvi/utils.py,sha256=izQcImnc3aRyhHwCyOlRxKxrMKBfMvc3CnssehhvX6c,2271
-arvi/data/info.svg,sha256=0IMI6W-eFoTD8acnury79WJJakpBwLa4qKS4JWpsXiI,489
-arvi/data/obs_affected_ADC_issues.dat,sha256=pbwzvRBjS8KsLrKi1Cdgfihvo3hK3FUN8Csdysw-vAw,10653
-arvi/data/obs_affected_blue_cryostat_issues.dat,sha256=z4AK17xfz8tGTDv1FjRvQFnio4XA6PNNfDXuicewHk4,1771
-arvi-0.1.8.dist-info/LICENSE,sha256=6JfQgl7SpM55t0EHMFNMnNh-AdkpGW25MwMiTnhdWQg,1068
-arvi-0.1.8.dist-info/METADATA,sha256=glmaHwWQqBB9ehRyZhLreP1Y9lRB_YSpnhAx7m4Mpk0,1305
-arvi-0.1.8.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-arvi-0.1.8.dist-info/top_level.txt,sha256=4EeiKDVLD45ztuflTGfQ3TU8GVjJg5Y95xS5XjI-utU,5
-arvi-0.1.8.dist-info/RECORD,,