o e%@sddlZddlZddlZddlZddlmZddlmZddlZddlm Z ddl m Z m Z m Z mZddlmZddlmZddlmZmZdd lmZd Zdd d Zd dZdddZdadddZddZdddZGdddZ dS)N)getpass)urlparse)rfapi)API_URLAPP_URL DEMO_KEYSload_roboflow_api_key)Project) Workspace) CLIPModel GazeModel) write_linez1.1.16cCst|tstdttddtt|tdd|Dr`|tvr&|Sttd|}|j dkr9t|j |j dkrZ|d krVt d t d |d 7}t||||Std |}|Stjd td|d|ddS)Nz.API Key is of Incorrect Type Expected Type: z Input Type: css|] }|r|VqdSN)islower).0cr@/var/www/myenv/lib/python3.10/site-packages/roboflow/__init__.py szcheck_key..z /?api_key=iz retrying...z?There was an error validating the api key with Roboflow server.z8upload and label your dataset, and get an API KEY here: z/?model=z&ref= onboarding) isinstancestr RuntimeErrortypeanyrrequestspostr status_codetextprinttimesleep check_keyjsonsysstdoutwriter)api_keymodelnotebook num_retriesresponserrrrr'sF      r'cCst|}|d}t||S)N workspace)r'Roboflow)r,r1wrrrauth>s r5Fc Cstj}|dkrtjtdd}n tjtdd}tjd|d}tj|r1|s1tddStj|r>|r>t||durKtd td n td td |d t d }t td|}|j dkr| }|durutdtjtj|sttj|d|i}t|dd}|d|}|d|d<t|d}t j||ddWddS1swYdS|dS)Nnt USERPROFILEzroboflow/config.jsonHOMEz.config/roboflow/config.jsonROBOFLOW_CONFIG_DIRdefaultz_You are already logged into Roboflow. To make a different login,run roboflow.login(force=True).zvisit z+/auth-cli to get your authentication token.z/auth-cli/?workspace=z" to get your authentication token.z%Paste the authentication token here: z/query/cliAuthToken/rz9Invalid API key. Please check your API key and try again. workspacesrurl RF_WORKSPACEr4indent)osnamepathjoingetenvisfiler removerrr getr"r( ValueErrorexistsdirnamemkdirlistkeysopendumpraise_for_status) r2forceos_name default_path conf_locationtokenr_logindefault_workspace_idfrrrloginEsH      r[cCsRtjdtddd}tj|std|dur!tatSt|atS)zHigh level function to initialize Roboflow. Args: the_workspace: the workspace url to initialize. If None, the default workspace will be used. Returns: None r9r8z/.config/roboflow/config.jsonr:z?To use this method, you must first login - run roboflow.login()N)rBrFrDrGrr3r2active_workspace) the_workspacerVrrrinitialize_roboflows     r^cCsbt}d|vs d|vr t|}|jd}|d}t|d}nd||}||}|j}|S)zHigh level function to load Roboflow models. Args: model_url: the model url to load. Must be from either app.roboflow.com or universe.roboflow.com Returns: the model object to use for inference universe.roboflow.comapp.roboflow.com/r?GModel URL must be from either app.roboflow.com or universe.roboflow.com)r^rrDsplitintprojectversionr-) model_urloperate_workspace parsed_url path_partsrfrgr-rrr load_models    rlc Cspd|vsd|vr!t|}|jd}|d}t|d}|d}ndt|d}||}||}|||S) aHigh level function to download data from Roboflow. Args: dataset_url: the dataset url to download. Must be from either app.roboflow.com or universe.roboflow.com model_format: the format the dataset will be downloaded in location: the location the dataset will be downloaded to Returns: The dataset object with location available as dataset.location r_r`rar?rbrrc)r])rrDrdrer^rfrgdownload) dataset_url model_formatlocationrjrkrfrgr]rirrrdownload_datasets        rqc@s@eZdZ   d ddZddZdddZdd d Zd d ZdS)r3N undefinedcCs6||_|jdur t|_||_||_d|_|dSNF)r,rror.rr5)selfr,ror.rrr__init__s  zRoboflow.__init__cCsHt|j|j|j}|dkrd|_|S|tvrd|_|S|d}||_|S)NrTr2)r'r,ror.rruniversecurrent_workspace)rtr1r4rrrr5sz Roboflow.authcCsztjdtjdtj|dur|j}|jtvr&ti|j||jSt |}|p.|j}t ||}t||||jS)Nz loading Roboflow workspace...r) r)r*r+flushrwr,rr rorr get_workspace)rtr]workspace_api_keyr, list_projectsrrrr2s      zRoboflow.workspacecCs|durd|vr|d}|d|d}}n|j}ttd|d|d|j}|jdkr6t|j| d}t |j|S)zFunction that takes in the name of the project and returns the project object :param project_name api_key: project name :param the_workspace workspace name :return project object Nrarrz ?api_key=rrf) rsplitrwr rIrr,r"rr#r(r )rt project_namer]splitted_project dataset_inforrrrfs $    zRoboflow.projectcCs|j|jd}tj|ddS)zto string function)r,r2r?r@)r,r2r(dumps)rt json_valuerrr__str__szRoboflow.__str__)Nrrrrr)__name__ __module__ __qualname__rur5r2rfrrrrrr3s    r3)rrsr)!r(rBr)r%r urllib.parserr roboflow.adaptersrroboflow.configrrrrroboflow.core.projectr roboflow.core.workspacer roboflow.modelsr r roboflow.util.generalr __version__r'r5r[r\r^rlrqr3rrrrs,       * ;