behavior only NWB file dev

behavior_only.py

from allensdk.internal.api.mtrain_api import MtrainApi
from allensdk.internal.api.behavior_lims_api import BehaviorLimsApi
from allensdk.brain_observatory.behavior.behavior_session import BehaviorSession
from allensdk.brain_observatory.behavior.behavior_api.behavior_nwb_api import BehaviorNwbApi


bl_api = BehaviorLimsApi()
m_api = MtrainApi()

# Get a session:
labtracks_id_list = m_api.get_subjects()
foraging_id = m_api.get_behavior_training_df(LabTracks_ID=463810)['behavior_session_uuid'].values[0]
behavior_session_id = bl_api.foraging_id_to_behavior_session_id(foraging_id)
session = BehaviorSession.from_lims(behavior_session_id)

# Write a file:
BehaviorNwbApi('tmp.nwb').save(session)