
import os
import numpy as np
from astropy.time import Time

from pyuvdata import UVData

from hera_mc import mc

a = mc.get_mc_argument_parser()
a.description = """Read the obsid from a file and create a record in M&C."""
a.add_argument('files', metavar='file', type=str, nargs='*', default=[],
               help='*.uv files to extract')
args = a.parse_args()
db = mc.connect_to_mc_db(args)


for uvfile in args.files:
    # assume our data file is uvh5
    uv = UVData()
    uv.read_uvh5(uvfile, read_data=False)
    times = np.unique(uv.time_array)
    starttime = Time(times[0], scale='utc', format='jd')
    stoptime = Time(times[-1], scale='utc', format='jd')
    obsid = int(np.floor(starttime.gps))

    with db.sessionmaker() as session:
        obs = session.get_obs(obsid)
        if len(obs) > 0:
            print("observation {obs} already in M&C, skipping".format(obs=obsid))
            continue
        print("Inserting obsid into M&C:" + str(obsid))

        session.add_obs(starttime, stoptime, obsid)
    session.commit()


import numpy as np
from astropy.time import Time

from pyuvdata import UVData

from hera_mc import mc

a = mc.get_mc_argument_parser()
a.description = """Read the obsid from a file and create a record in M&C."""
a.add_argument('files', metavar='file', type=str, nargs='*', default=[],
               help='*.uvh5 files to add')
args = a.parse_args()
db = mc.connect_to_mc_db(args)


for uvfile in args.files:
    # assume our data file is uvh5
    uv = UVData()
    uv.read_uvh5(uvfile, read_data=False)
    times = np.unique(uv.time_array)
    starttime = Time(times[0], scale='utc', format='jd')
    stoptime = Time(times[-1], scale='utc', format='jd')
    obsid = int(np.floor(starttime.gps))

    with db.sessionmaker() as session:
        obs = session.get_obs(obsid)
        if len(obs) > 0:
            print("observation {obs} already in M&C, skipping".format(obs=obsid))
            continue
        print("Inserting obsid into M&C:" + str(obsid))

        session.add_obs(starttime, stoptime, obsid)
    session.commit()

Fix flake8 issue and fix up documentation
Fix flake8 issue and fix up documentation

  
- import os
  import numpy as np
  from astropy.time import Time
  
  from pyuvdata import UVData
  
  from hera_mc import mc
  
  a = mc.get_mc_argument_parser()
  a.description = """Read the obsid from a file and create a record in M&C."""
  a.add_argument('files', metavar='file', type=str, nargs='*', default=[],
-                help='*.uv files to extract')
+                help='*.uvh5 files to add')
  args = a.parse_args()
  db = mc.connect_to_mc_db(args)
  
  
  for uvfile in args.files:
      # assume our data file is uvh5
      uv = UVData()
      uv.read_uvh5(uvfile, read_data=False)
      times = np.unique(uv.time_array)
      starttime = Time(times[0], scale='utc', format='jd')
      stoptime = Time(times[-1], scale='utc', format='jd')
      obsid = int(np.floor(starttime.gps))
  
      with db.sessionmaker() as session:
          obs = session.get_obs(obsid)
          if len(obs) > 0:
              print("observation {obs} already in M&C, skipping".format(obs=obsid))
              continue
          print("Inserting obsid into M&C:" + str(obsid))
  
          session.add_obs(starttime, stoptime, obsid)
      session.commit()
  
Fix flake8 issue and fix up documentation
## Code Before:

import os
import numpy as np
from astropy.time import Time

from pyuvdata import UVData

from hera_mc import mc

a = mc.get_mc_argument_parser()
a.description = """Read the obsid from a file and create a record in M&C."""
a.add_argument('files', metavar='file', type=str, nargs='*', default=[],
               help='*.uv files to extract')
args = a.parse_args()
db = mc.connect_to_mc_db(args)


for uvfile in args.files:
    # assume our data file is uvh5
    uv = UVData()
    uv.read_uvh5(uvfile, read_data=False)
    times = np.unique(uv.time_array)
    starttime = Time(times[0], scale='utc', format='jd')
    stoptime = Time(times[-1], scale='utc', format='jd')
    obsid = int(np.floor(starttime.gps))

    with db.sessionmaker() as session:
        obs = session.get_obs(obsid)
        if len(obs) > 0:
            print("observation {obs} already in M&C, skipping".format(obs=obsid))
            continue
        print("Inserting obsid into M&C:" + str(obsid))

        session.add_obs(starttime, stoptime, obsid)
    session.commit()

## Instruction:
Fix flake8 issue and fix up documentation
## Code After:

import numpy as np
from astropy.time import Time

from pyuvdata import UVData

from hera_mc import mc

a = mc.get_mc_argument_parser()
a.description = """Read the obsid from a file and create a record in M&C."""
a.add_argument('files', metavar='file', type=str, nargs='*', default=[],
               help='*.uvh5 files to add')
args = a.parse_args()
db = mc.connect_to_mc_db(args)


for uvfile in args.files:
    # assume our data file is uvh5
    uv = UVData()
    uv.read_uvh5(uvfile, read_data=False)
    times = np.unique(uv.time_array)
    starttime = Time(times[0], scale='utc', format='jd')
    stoptime = Time(times[-1], scale='utc', format='jd')
    obsid = int(np.floor(starttime.gps))

    with db.sessionmaker() as session:
        obs = session.get_obs(obsid)
        if len(obs) > 0:
            print("observation {obs} already in M&C, skipping".format(obs=obsid))
            continue
        print("Inserting obsid into M&C:" + str(obsid))

        session.add_obs(starttime, stoptime, obsid)
    session.commit()
