jin-zhe / gopro Goto Github PK

def extract_all(self):
    self.extract_gpx()
    self.extract_json()
    self.extract_metadata()

  def extract_gpx(self):
    gpx_path = os.path.join(self.video_dir, self.filename + '.gpx')

    # If reprocessing or gpx file does not yet exists
    if self.reprocess or not os.path.isfile(gpx_path):
      command = [
        self.gopro2gpx_path,
        '-i', self.telemetry_path,
        '-o', gpx_path
      ]
      GoProTelemetry.call_subprocess(command)

  def extract_json(self):
    json_path = os.path.join(self.video_dir, self.filename + '.json')
    # If reprocessing or json file does not yet exists
    if self.reprocess or not os.path.isfile(json_path):
      command = [
        self.gopro2json_path,
        '-i', self.telemetry_path,
        '-o', json_path
      ]
      GoProTelemetry.call_subprocess(command)

  def extract_metadata(self):
    gps_path = os.path.join(self.video_dir, self.filename + '_gps.csv')
    gyro_path = os.path.join(self.video_dir, self.filename + '_gyro.csv')
    accl_path = os.path.join(self.video_dir, self.filename + '_accl.csv')
    temp_path = os.path.join(self.video_dir, self.filename + '_temp.csv')
    
    # If reprocessing or none of the metadata files yet exists
    if self.reprocess or not (

      os.path.isfile(gps_path) and
      os.path.isfile(gyro_path) and
      os.path.isfile(accl_path) and
      os.path.isfile(temp_path)):

      command = [
        self.gpmdinfo_path,
        '-i', self.telemetry_path
      ]
      GoProTelemetry.call_subprocess(command)
      
      # Rename and move files into video directory
      print(self.gpmdinfo_path)
      shutil.move('./gps.csv', gps_path)
      shutil.move('./gyro.csv', gyro_path)
      shutil.move('./accl.csv', accl_path)
      shutil.move('./temp.csv', temp_path)```