In [ ]:
from os import path

from astroquery.vizier import Vizier
import astropy.units as u
import numpy as np
import matplotlib.pyplot as plt
plt.style.use('apw-notebook')
%matplotlib inline

from comoving_rv.db import Session, Base, db_connect
from comoving_rv.db.model import (Run, Observation, TGASSource, SimbadInfo,
                                  SpectralLineInfo, SpectralLineMeasurement, PriorRV)
from comoving_rv.db import get_best_rv

In [ ]:
base_path = '/Volumes/ProjectData/gaia-comoving-followup/'
db_path = path.join(base_path, 'db.sqlite')
engine = db_connect(db_path)
session = Session()

In [ ]:
observations = session.query(Observation).filter(Observation.prior_rv_id == None).all()

In [ ]:
done = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49]
for obs in observations:
    if obs.id in done:
        continue
        
    print(obs)
    result = get_best_rv(obs)
    if result is not None:
        print("\t", result[0])
        rv, rv_err, rv_qual, rv_bibcode, rv_source = result

        prv = PriorRV(rv=rv*u.km/u.s, err=rv_err*u.km/u.s,
                      qual=rv_qual, bibcode=rv_bibcode,
                      source=rv_source)
        obs.prior_rv = prv
        session.add(prv)
        session.commit()
        
        done.append(obs.id)
    
    done.append(obs.id)

In [ ]:
print(done)

In [ ]:
obs.simbad_info

In [ ]:
obs.ra.degree, obs.dec.degree

In [ ]:
obs.tgas_source.source_id

In [ ]:
from astropy.table import Table
tgas = Table.read('../../gaiadr1/data/stacked_tgas.fits')

In [ ]:
tgas[tgas['source_id'] == 832479767088738432]

In [ ]:


In [ ]:
prvs = u.Quantity([x[0] for x in session.query(PriorRV.err).all()])

In [ ]:
plt.hist(prvs, bins='auto');

In [ ]: