76 lines
13 KiB
Plaintext
76 lines
13 KiB
Plaintext
|
a
|
|||
|
|
Ïœ>jÕ9 ã @ s d Z ddlmZmZmZmZmZ ddlmZ ddl m
|
|||
|
|
Z
|
|||
|
|
m Z ddl m
|
|||
|
|
Z
|
|||
|
|
ddlmZmZmZ ddlmZmZ ddlZed d
|
|||
|
|
d ee
|
|||
|
|
d d
|
|||
|
|
�ƒgd�Zejedgdgdgd� ddlZej dd¡ ej dd¡ ej dd¡ ej dd¡ ej dd¡ ej dd¡ ddlmZm Z m Z ddl m Z m Z ddl!m"Z"m#Z# ddl$m%Z%m&Z& ddl'm(Z( dd l)m*Z*m+Z+m,Z, dd l-m.Z. dd l/m0Z0 eƒ Z1e"ƒ Z2e%ƒ Z3e(ƒ Z4e.d d �Z5e0ƒ Z6d!d"„ Z7e,d#œd$d%„Z8G d&d'„ d'ƒZ9e :d(¡ed)œd*d+„ƒZ;e :d,¡ed)œd-d.„ƒZ<e =d/¡ee*j>ƒfe?ed0œd1d2„ƒZ@e =d3¡ee*j>ƒfe?ed0œd4d5„ƒZAe :d6¡dee*j>ƒfe?e?e?e?ee? ed7œd8d9„ƒZBe =d:¡ee*j>ƒfed;œd<d=„ƒZCe =d>¡ee*j>ƒfe?ed?œd@dA„ƒZDe =dB¡ee*j>ƒfe?e?edCœdDdE„ƒZEe :dF¡ee*j>ƒfe edGœdHdI„ƒZFe =dJ¡ee*j>ƒfe?edKœdLdM„ƒZGe =dN¡edOdPdQdR�edddS�ee*j>ƒfe?eHeHedTœdUdV„ƒZIe =dW¡ee*j>ƒfe?ed0œdXdY„ƒZJe =dZ¡dd[ee*j>ƒfee? e?ed\œd]d^„ƒZKe =d_¡d[ee*j>ƒfe?e?ed`œdadb„ƒZLe =dc¡d[ee*j>ƒfe?e?eddœdedf„ƒZMe :dg¡ee*j>ƒfee? edhœdidj„ƒZNe :dk¡dlee*j>ƒfe&e?edmœdndo„ƒZOe :dp¡dlee*j>ƒfeHee? e?edqœdrds„ƒZPe =dt¡ee*j>ƒfe?eduœdvdw„ƒZQe :dx¡dydyee*j>ƒfe?e?eRee?e?edzœd{d|„ƒZSe :d}¡dydydydyee*j>ƒfe?eRee?e?e?e?ed~œdd€„ƒZTe :d�¡ee*j>ƒfe?e?eReHeed‚œdƒd„„ƒZUe :d…¡ed)œd†d‡„ƒZVe :dˆ¡ed)œd‰dŠ„ƒZWe :d‹¡ed)œdŒd�„ƒZXe =dŽ¡d�d�„ ƒZYeZd‘k�r¾ddl[Z[e[j\ed’d“d”� dS )•zf
|
|||
|
|
IOM API - FastAPI application for Infrastructure Object Model
|
|||
|
|
Combines all layers into a unified API
|
|||
|
|
é )ÚFastAPIÚ
|
|||
|
|
HTTPExceptionÚQueryÚDependsÚRequest)ÚCORSMiddleware)Ú
|
|||
|
|
HTTPBearerÚ HTTPAuthorizationCredentials)Ú JSONResponse)ÚListÚOptionalÚDict)ÚdatetimeÚdateNzIOM APIz Infrastructure Object Model APIú1.0.0F)Z
|
|||
|
|
auto_error)ÚtitleÚ descriptionÚversionZ dependenciesÚ*)Z
|
|||
|
|
allow_originsZ
|
|||
|
|
allow_methodsZ
|
|||
|
|
allow_headersz(/home/bernt/.openclaw/workspace/iom/corez//home/bernt/.openclaw/workspace/iom/observationz*/home/bernt/.openclaw/workspace/iom/defectz(/home/bernt/.openclaw/workspace/iom/riskz*/home/bernt/.openclaw/workspace/iom/ledgerz6/home/bernt/.openclaw/workspace/iom/visual_geolocation)Ú
|
|||
|
|
GOIDGeneratorÚ
|
|||
|
|
validate_goidÚ
|
|||
|
|
parse_goid)Ú ObservationÚObservationSummary)ÚDefectRegistryÚDefectCategory)ÚRiskCalculatorÚ
|
|||
|
|
RiskScores)ÚIOMLedgerMapper)Ú auth_managerÚUserRoleÚ
|
|||
|
|
Permission)ÚVisualGeolocationPipeline)ÚTemporalAnalyzerT)Z use_real_aic C s
|
|||
|
|
t tjƒS ©N)r r Úget_current_user© r& r& ú//home/bernt/.openclaw/workspace/iom/api/main.pyÚ require_auth: s r( ©Z
|
|||
|
|
permissionc C s t t | ¡ƒS r$ )r r Úrequire_permissionr) r& r& r' r* = s r* c @ s e Zd Zdeeedœdd„ZdS )Ú AuthRequestÚreadonly©ÚusernameÚpasswordÚrolec C s || _ || _|| _d S r$ r- )Úselfr. r/ r0 r& r& r' Ú__init__D s zAuthRequest.__init__N)r, )Ú__name__Ú
|
|||
|
|
__module__Ú __qualname__Ústrr2 r& r& r& r' r+ C s r+ z/auth/register)Úrequestc à s| | d¡}| d¡}| dd¡}|r(|s4tddd�‚z t|ƒ}W n$ tyd tdd|› �d�‚Y n0 t ||¡}||d œS )
|
|||
|
|
zRegister a new userr. r/ r0 r, é� ú Username and password required©Ú status_codeZdetailzInvalid role: )Útokenr0 )Úgetr r Ú
|
|||
|
|
ValueErrorr Ú create_token)r7 r. r/ r0 Z user_roler< r& r& r' Ú
|
|||
|
|
register_userI s
|
|||
|
|
|
|||
|
|
r@ z /auth/loginc à s> | d¡}| d¡}|r |s(tddd�‚t |tj¡}d|iS )zLogin and get tokenr. r/ r8 r9 r: r< )r= r r r? r ZREADONLY)r7 r. r/ r< r& r& r' Úlogin] s
|
|||
|
|
|
|||
|
|
rA z/goid/validate/{goid}©ÚgoidÚuserc à s t | ƒ\}}||dœS )zValidate a GOID)ZvalidÚerror)r )rC rD Zis_validrE r& r& r' Úvalidate_goid_endpointm s rF z/goid/parse/{goid}c
|
|||
|
|
à sH zt | ƒ}|W S tyB } z tdt|ƒd�‚W Y d}~n
|
|||
|
|
d}~0 0 dS )z Parse a GOID into componentsr8 r: N)r r> r r6 )rC rD ÚparsedÚer& r& r' Úparse_goid_endpointt s
|
|||
|
|
rI z/goid/generate)ÚdomainÚsystemÚ subsystemÚobj_typeÚ
|
|||
|
|
location_hashrD c
|
|||
|
|
à sV z t | ||||¡}d|iW S tyP } z tdt|ƒd�‚W Y d}~n
|
|||
|
|
d}~0 0 dS )zGenerate a new GOIDrC r8 r: N)Úgoid_genZgenerater> r r6 )rJ rK rL rM rN rD rC rH r& r& r' Ú
|
|||
|
|
generate_goid~ s
|
|||
|
|
|
|||
|
|
|
|||
|
|
rP z/taxonomy/domains©rD c à s t ¡ S )zList all domains)rO Ú list_domainsrQ r& r& r' rR � s rR z"/taxonomy/domains/{domain}/systems)rJ rD c
|
|||
|
|
à sF z t | ¡W S ty@ } z tdt|ƒd�‚W Y d}~n
|
|||
|
|
d}~0 0 dS )zList systems in a domainé” r: N)rO Ú list_systemsr> r r6 )rJ rD rH r& r& r' rT • s rT z3/taxonomy/domains/{domain}/systems/{system}/objects)rJ rK rD c
|
|||
|
|
à sH zt | |¡W S tyB } z tdt|ƒd�‚W Y d}~n
|
|||
|
|
d}~0 0 dS )zList objects in a systemrS r: N)rO Ú list_objectsr> r r6 )rJ rK rD rH r& r& r' rU ž s rU z
|
|||
|
|
/observations©Z observationrD c à s" | j pdt ¡ j› d�d| jdœS )zCreate a new observationzOBS-z-0000001Úcreated)ÚidÚstatusÚ object_goid)rX r ÚnowÚyearrZ rV r& r& r' Úcreate_observation© s ýr] z /observations/{observation_id}©Úobservation_idrD c à s t ddd�‚dS )zGet observation by IDrS zObservation not foundr: N)r r^ r& r& r' Úget_observation´ s r` z /objects/{goid}/observationséd é iè )ÚgeÚle)rc ©rC ÚlimitÚoffsetrD c à s | g ddœS )z Get observations for an objectr )rZ Ú observationsÚtotalr& re r& r& r' Úget_object_observations» s
|
|||
|
|
ýrj z/objects/{goid}/summaryc à s | dddœS )z%Get observation summary for an objectr N)rZ Zobservation_countZlatest_conditionr& rB r& r& r' Úget_object_summaryË s ýrk z/defectsÚsv)ÚcategoryÚlangrD c ƒ sb | rBzt | ƒ}t |¡}W qP ty> tdd| › �d�‚Y qP0 nttj ¡ ƒ}‡ fdd„|D ƒS )zList all defect codesr8 zInvalid category: r: c s g | ]}| ˆ ¡‘qS r& ©Úto_dict©Ú.0Úcode©rn r& r' Ú
|
|||
|
|
<listcomp>è ó z list_defects.<locals>.<listcomp>)r Ú
|
|||
|
|
defect_regZget_by_categoryr> r ÚlistZ_codesÚvalues)rm rn rD ÚcatZcodesr& rt r' Ú list_defectsØ s r{ z/defects/{code})rs rn rD c à s* t | ¡}|s tdd| › �d�‚| |¡S )zGet defect code by coderS zDefect code not found: r: )rw r= r rp )rs rn rD Zdefectr& r& r' Ú
|
|||
|
|
get_defectë s
|
|||
|
|
r| z/defects/search/{query})Úqueryrn rD c ƒ s t | ˆ ¡}‡ fdd„|D ƒS )zSearch defect codesc s g | ]}| ˆ ¡‘qS r& ro rq rt r& r' ru ø rv z"search_defects.<locals>.<listcomp>)rw Úsearch)r} rn rD Úresultsr& rt r' Úsearch_defectsô s r€ z/defects/suggest)ÚkeywordsrD c à s t | ¡}dd„ |D ƒS )z)Suggest defect codes based on AI keywordsc S s g | ]\}}|j |j|d œ‘qS ))rs ÚnameÚ
|
|||
|
|
confidence)rs Zname_en)rr rs rƒ r& r& r' ru ÿ s
|
|||
|
|
üýz#suggest_defects.<locals>.<listcomp>)rw Zget_ai_suggestions)r� rD Z suggestionsr& r& r' Úsuggest_defectsû s
|
|||
|
|
úr„ z/risk/calculateÚdefault)ÚscoresÚ object_typerD c à s t | |¡}|S )zCalculate risk score)Ú risk_calcZ calculate)r† r‡ rD Úresultr& r& r' Úcalculate_risk s rŠ z /risk/calculate-from-observation)Ú conditionÚ defect_codesr‡ rD c à s t | ||¡}|S )z$Calculate risk from observation data)rˆ Zcalculate_from_observation)r‹ rŒ r‡ rD r‰ r& r& r' Ú calculate_risk_from_observation s r� z/risk/weights/{object_type})r‡ rD c à s t | ¡}| |dœS )z Get risk weights for object type)r‡ Úweights)rˆ Zget_weights_for_type)r‡ rD rŽ r& r& r' Úget_risk_weights" s
|
|||
|
|
r� z/ledger/installationÚ )rC r‡ ÚcostÚinstallation_dateÚ
|
|||
|
|
contractorÚprojectrD c à s( t | |||||¡}| dd„ |D ƒdœS )z'Map installation to ledger transactionsc S s g | ] }| ¡ ‘qS r& ro ©rr Útr& r& r' ru ; rv z$map_installation.<locals>.<listcomp>©rC Ú transactions)Ú
|
|||
|
|
ledger_mapperÚmap_installation)rC r‡ r‘ r’ r“ r” rD r˜ r& r& r' rš + s ÿ þrš z/ledger/maintenance)rC r‘ Úmaintenance_dater_ Ú defect_coder“ r rD c à s* t | ||||||¡}| dd„ |D ƒdœS )z&Map maintenance to ledger transactionsc S s g | ] }| ¡ ‘qS r& ro r• r& r& r' ru P rv z#map_maintenance.<locals>.<listcomp>r— )r™ Úmap_maintenance) rC r‘ r› r_ rœ r“ r rD r˜ r& r& r' r� ? s ÿ þr� z/ledger/depreciation)rC r‡ Úacquisition_valueÚ useful_lifeÚdepreciation_daterD c à s& t | ||||¡}| dd„ |D ƒdœS )z'Map depreciation to ledger transactionsc S s g | ] }| ¡ ‘qS r& ro r• r& r& r' ru c rv z$map_depreciation.<locals>.<listcomp>r— )r™ Úmap_depreciation)rC r‡ rž rŸ r rD r˜ r& r& r' r¡ T s
|
|||
|
|
|
|||
|
|
ÿ þr¡ z/visual-geolocation/analyzec
|
|||
|
|
à sÀ | d¡}| d¡}|s$tddd�‚zbt ||¡}|p8d|j|j|j|jdœ|j|j |j
|
|||
|
|
j |j
|
|||
|
|
j t
|
|||
|
|
|j
|
|||
|
|
jƒt
|
|||
|
|
|j
|
|||
|
|
jƒdœ|j|jd œW S tyº } z td
|
|||
|
|
t|ƒd�‚W Y d }~n
|
|||
|
|
d }~0 0 d S ) z*Analyze image and extract evidence packageÚ
|
|||
|
|
image_pathÚimage_idr8 zimage_path requiredr: Úunknown)ÚlatÚlngÚaccuracyrƒ )ZoverallÚuncertainty_radiusZsupporting_evidence_countZ contradicting_evidence_count)r£ ÚpositionÚmethodÚevidence_summaryÚconfidence_reportÚ map_matchesÚsimilar_imagesiô N)r= r Ú
|
|||
|
|
visual_geoÚ
|
|||
|
|
process_imager¥ r¦ r§ rƒ rª r« r¬ Zoverall_confidencer¨ ÚlenZsupporting_evidenceZcontradicting_evidencer r® Ú Exceptionr6 )r7 r¢ r£ r‰ rH r& r& r' Ú
|
|||
|
|
analyze_imagei s0
|
|||
|
|
|
|||
|
|
ü
|