o PcH@s~dZddlZddlZddlZddlmZmZddlmZmZm Z m Z ddl Z ddl ZddlZGdddZGdddeZdS) zSClasses to convert FIT files to GPX, including tools to process Strava Bulk Export N)datetime timedelta)DictUnionOptionalTuplec @seZdZdZddefddZdejjde e e e e eefffdd Zdejjdee e e e ee e fffd d Zd e deejejffd dZ  dddZddZddZdS) ConverterzOMain converter that holds the FIT > pd.DataFrame and pd.DataFrame > GPX methodsF status_msgcCs||_gd|_gd|_dS)zMain constructor for StravaConverter Parameters: status_msg (bool): Option to have the Converter print to console with status messages, such as number of files converted. ) latitude longitudelap timestampaltitudeenhanced_altitude temperature heart_ratecadencespeedenhanced_speedpower)number start_timetotal_distancetotal_elapsed_time max_speedmax_heart_rateavg_heart_rateN)r _colnames_points_colnames_laps)selfr r =/var/www/html/gps/gps/lib/python3.10/site-packages/fit2gpx.py__init__s zConverter.__init__framereturncCs4i}|jddD]}||r||||<q |S)zWExtract some data from a FIT frame representing a lap and return it as a dict. N)r has_field get_valuerr#datafieldr r r! _get_fit_laps6s  zConverter._get_fit_lapscCsi}|dr |dsdS|ddur|ddurdS|dd|d<|dd|d<|jddD]}||rE||||<q7|S)z`Extract some data from an FIT frame representing a track point and return it as a dict. position_lat position_longNgllfAr r )r&r'rr(r r r!_get_fit_pointsEs zConverter._get_fit_pointsfnamec Cs tj|d}|dkrtjdg}g}d}t|C}|D]8}t|tj j rZ|j dkrC| |}|durB||d<| |q"|j dkrZ||} || d<| | |d7}q"Wdn1sewYtj||jd} | jdd d tj||jd} | | fS) a<Takes the path to a FIT file and returns two Pandas DataFrames for lap data and point data Parameters: fname (str): string representing file path of the FIT file Returns: dfs (tuple): df containing data about the laps , df containing data about the individual points. r%.fitInput file must be a .FIT file.recordNr rcolumnsTinplace)ospathsplitextlower fitdecode exceptionsFitHeaderError FitReader isinstancerecordsFitDataMessagenamer/appendr+pd DataFramer set_indexr) rr0input_extension data_points data_lapslap_nofit_filer#single_point_datasingle_lap_datadf_laps df_pointsr r r!fit_to_dataframes^s6         zConverter.fit_to_dataframesr r Nc  s$||g} |r | ||r| |tfdd| Dr!tdtj} tj} | j| tj} | j | || jd_ | | jd_ t |sO|nd| jd_|| jd_jD]1}tjjj||fj||f|r{t j||fnd|rj||fndd}| j|q^| S)a Convert a pandas dataframe to gpx Parameters: df_points (pd.DataFrame): pandas dataframe containing at minimum lat and long info of points col_alt (str): name of the altitudes column col_time (str): name of the time column col_long (str): name of the longitudes column col_lat (str): name of the latitudes column gpx_name (str): name for the gpx track (note is not the same as the file name) gpx_desc (str): description for the gpx track gpx_type : activity type for the gpx track (can be str, or int) gpx_link (str): link to the gpx activity c3s|]}|jvVqdSNr4).0elemrPr r! sz-Converter.dataframe_to_gpx..zThe input dataframe must consist of point coordinates in longitude and latitude. Ideally, it should be the df_points output from the fit_to_dataframes() method.rN)r r time elevation)rDanyKeyErrorgpxpygpxGPXGPXTracktracksGPXTrackSegmentsegmentsrCtyperEisna descriptionlinkindex GPXTrackPointloc Timestamppoints)rrPcol_latcol_longcol_timecol_altgpx_namegpx_descgpx_linkgpx_type cols_to_checkr\ gpx_track gpx_segmentidx track_pointr rUr!dataframe_to_gpxs2             zConverter.dataframe_to_gpxc Cstj|d}|dkrtdtj|d}|dkr td||\}}ddg}|D]#}||dkrP|d |dkrP||j|d |d d q-|j|d d ddd} t |d} | | Wd| S1sswY| S)zMethod to convert a FIT file into a GPX file Parameters: f_in (str): file path to FIT activity f_out (str): file path to save the converted FIT file r%r1r2.gpxz Output file must be a .gpx file.rrr enhanced_Tr6r r r rPrkrlrmrnwN) r8r9r: Exception TypeErrorrQcountfillnarxopenwriteto_xml) rf_inf_outrHoutput_extensionrOrPenhanced_fieldsr*r\fr r r! fit_to_gpxs2&  zConverter.fit_to_gpxcCs|ddkr |d7}|r|ddkr|d7}tj|s!t|ddt|D}|D]}|j|||tj|dddq-|jrQtt |dd Sd S) zMethod to convert all FIT files in a directory to GPX files Parameters: dir_in (str): path to directory with all FIT activities dir_out (str): path to directory to save the converted FIT files to /cSg|] }d|vr|qSr1r;rSrr r r! z-Converter.fit_to_gpx_bulk..rry)rrz" files converted from .fit to .gpxN) r8r9isdirmkdirlistdirrr:r printlen)rdir_indir_out fit_files f_activityr r r!fit_to_gpx_bulks   zConverter.fit_to_gpx_bulk)F)r r NNNNNN)__name__ __module__ __qualname____doc__boolr"r<rArBrstrrfloatrrintr+rr/rrErFrQrxrrr r r r!rs %  ) 8 'rcs:eZdZdZd fdd ZddZddZd d ZZS) StravaConverterzRConverter to use when converting .FIT files from Strava data download to GPX filesNcst|ddkr|d7}|r|ddkr|d7}|s!|d}dt|vs0tj|ds4tdtj|s?t|||_||_ |jd|_ dS) a1Main constructor for StravaConverter Parameters: dir_in (str): path to main strava data download folder dir_out (str): path to directory to save the converted FIT files to. If not provided, saves to an 'activities_gpx' directory in the main directory rrzactivities_gpx/zactivities.csv activitieszThe input directory must be the main Strava data download directory, i.e. must contain activities.csv and have a sub directory named "activities".z activities/N) superr"r8rr9rr}r_dir_in_dir_out_dir_activities)rrr __class__r r!r" s    zStravaConverter.__init__c stddtjD}fddtjD}|D]G}|dd}|tjvr-qt|d#}t|d}t||Wd n1sJwYWd n1sYwYt |qj rttj|kr|t t|d d St d d Sd S) zKMethod to unzip .gz files to their native format (e.g. gpx, tcx, fit, etc.)cSsh|] }|ddqS).r)splitrr r r! 1sz3StravaConverter.unzip_activities..csg|] }d|vrj|qS).gz)rrrr r!r4sz4StravaConverter.unzip_activities..rrbwbNz zipped files have been unzippedz0ERROR: Certain files have been deleted. Oopsies.) rr8rrreplacegziprshutil copyfileobjremover r)r cnt_activites zip_pathspath_zip path_unziprrr rr!unzip_activities.s$    z StravaConverter.unzip_activitiesc Csbt|jdd}ddt|jD}|D]}||j|\}}|j|dj |j dddf }|d}|d }|d } | sPt |d d }t| rXd} n | sdt | d d } ||d | d|d} |jd|ddddd| } |j|d} t| d} | | Wdn1swY|jrtt|dqdS)znMethod to convert all FIT files in a directory to GPX files, adding metadata from Strava to GPX files /activities.csvrcSrrrrr r r!rWrz5StravaConverter.strava_fit_to_gpx..FilenamerN Activity ID Activity NameActivity Descriptionasciiignore Activity Type"https://www.strava.com/activities/)rorrrprqr r r rr{ryr|zM files have been converted from .fit to .gpx files containing Strava metadatar )rEread_csvrrr8rrrQrhrcontainsilocto_dictisasciiencodercrxrrrrr rr)rdf_actsrrrOrPmdact_idact_nameact_desc strava_argsr\path_outrr r r!strava_fit_to_gpxNsH(  z!StravaConverter.strava_fit_to_gpxc svfddtjD}tjdd}|D]}|j|dj |j dddf }|d}|d }|d }| sHt| d d }t|rPd}n | s\t| d d }tj|d dd}t|} || jd_|d| jd_|| jd_d|| jd_j|d} t| d} | | Wdn1swYjrtt|dqdS)z[Method adds Strava metadata to default GPX files (i.e. files downloaded as GPX from Strava)cs(g|]}d|vr|tjvr|qS)ry)r8rrrrr r!rs(z7StravaConverter.add_metadata_to_gpx..rrrrNrrrrrrzutf-8)encodingrrryr|z+ .gpx files have had Strava metadata added.)r8rrrErrrrhrrrrrrrcrr[parser_rCrbrdrerrrr rr) r gpx_filesrgpx_pathrrrrf_gpxr\rrr rr!add_metadata_to_gpxs6(     z#StravaConverter.add_metadata_to_gpxrR) rrrrr"rrr __classcell__r r rr!r s ! ;r)rr8rrrrtypingrrrrpandasrE gpxpy.gpxr[r<rrr r r r!s}