app.py 219 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003400440054006400740084009401040114012401340144015401640174018401940204021402240234024402540264027402840294030403140324033403440354036403740384039404040414042404340444045404640474048404940504051405240534054405540564057405840594060406140624063406440654066406740684069407040714072407340744075407640774078407940804081408240834084408540864087408840894090409140924093409440954096409740984099410041014102410341044105410641074108410941104111411241134114411541164117411841194120412141224123412441254126412741284129413041314132413341344135413641374138413941404141414241434144414541464147414841494150415141524153415441554156415741584159416041614162416341644165416641674168416941704171417241734174417541764177417841794180418141824183418441854186418741884189419041914192419341944195419641974198419942004201420242034204420542064207420842094210421142124213421442154216421742184219422042214222422342244225422642274228422942304231423242334234423542364237423842394240424142424243424442454246424742484249425042514252425342544255425642574258425942604261426242634264426542664267426842694270427142724273427442754276427742784279428042814282428342844285428642874288428942904291429242934294429542964297429842994300430143024303430443054306430743084309431043114312431343144315431643174318431943204321432243234324432543264327432843294330433143324333433443354336433743384339434043414342434343444345434643474348434943504351435243534354435543564357435843594360436143624363436443654366436743684369437043714372437343744375437643774378437943804381438243834384438543864387438843894390439143924393439443954396439743984399440044014402440344044405440644074408440944104411441244134414441544164417441844194420442144224423442444254426442744284429443044314432443344344435443644374438443944404441444244434444444544464447444844494450445144524453445444554456445744584459446044614462446344644465446644674468446944704471447244734474447544764477447844794480448144824483448444854486448744884489449044914492449344944495449644974498449945004501450245034504450545064507450845094510451145124513451445154516451745184519452045214522452345244525452645274528452945304531453245334534453545364537453845394540454145424543454445454546454745484549455045514552455345544555455645574558455945604561456245634564456545664567456845694570457145724573457445754576457745784579458045814582458345844585458645874588458945904591459245934594459545964597459845994600460146024603460446054606460746084609461046114612461346144615461646174618461946204621462246234624462546264627462846294630463146324633463446354636463746384639464046414642464346444645464646474648464946504651465246534654465546564657465846594660466146624663466446654666466746684669467046714672467346744675467646774678467946804681468246834684468546864687468846894690469146924693469446954696469746984699470047014702470347044705470647074708470947104711471247134714471547164717471847194720472147224723472447254726472747284729473047314732473347344735473647374738473947404741474247434744474547464747474847494750475147524753475447554756475747584759476047614762476347644765476647674768476947704771477247734774477547764777477847794780478147824783478447854786478747884789479047914792479347944795479647974798479948004801480248034804480548064807480848094810481148124813481448154816481748184819482048214822482348244825482648274828482948304831483248334834483548364837483848394840484148424843484448454846484748484849485048514852485348544855485648574858485948604861486248634864486548664867486848694870487148724873487448754876487748784879488048814882488348844885488648874888488948904891489248934894489548964897489848994900490149024903490449054906490749084909491049114912491349144915491649174918491949204921492249234924492549264927492849294930493149324933493449354936493749384939494049414942494349444945494649474948494949504951495249534954495549564957495849594960496149624963496449654966496749684969497049714972497349744975497649774978497949804981498249834984498549864987498849894990499149924993499449954996499749984999500050015002500350045005500650075008500950105011501250135014501550165017501850195020502150225023502450255026502750285029503050315032503350345035503650375038503950405041504250435044504550465047504850495050505150525053505450555056505750585059506050615062506350645065506650675068506950705071507250735074507550765077507850795080508150825083508450855086508750885089509050915092509350945095509650975098509951005101510251035104510551065107510851095110511151125113511451155116511751185119512051215122512351245125512651275128512951305131513251335134513551365137513851395140514151425143514451455146514751485149515051515152515351545155515651575158515951605161516251635164516551665167516851695170517151725173517451755176517751785179518051815182518351845185518651875188518951905191519251935194519551965197519851995200520152025203520452055206520752085209521052115212521352145215521652175218521952205221522252235224522552265227522852295230523152325233523452355236523752385239524052415242524352445245524652475248524952505251525252535254525552565257525852595260526152625263526452655266526752685269527052715272527352745275527652775278527952805281528252835284528552865287528852895290529152925293529452955296529752985299530053015302530353045305530653075308530953105311531253135314531553165317531853195320532153225323532453255326532753285329533053315332533353345335533653375338533953405341534253435344534553465347534853495350535153525353535453555356535753585359536053615362536353645365536653675368536953705371537253735374537553765377537853795380538153825383538453855386538753885389539053915392539353945395539653975398539954005401540254035404540554065407540854095410541154125413541454155416541754185419542054215422542354245425542654275428542954305431543254335434543554365437543854395440544154425443544454455446544754485449545054515452545354545455545654575458545954605461546254635464546554665467546854695470547154725473547454755476547754785479548054815482548354845485548654875488548954905491549254935494549554965497549854995500550155025503550455055506550755085509551055115512551355145515551655175518551955205521552255235524552555265527552855295530553155325533553455355536553755385539554055415542554355445545554655475548554955505551
  1. from flask import Flask, jsonify, request, abort, send_from_directory
  2. from flask_cors import CORS
  3. from flask_socketio import SocketIO, emit
  4. import threading
  5. import queue
  6. import time
  7. import json
  8. import os
  9. import re
  10. import logging
  11. import uuid
  12. # 配置文件路径
  13. SERIAL_CONFIG_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'serial_config.json')
  14. MQTT_CONFIG_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'mqtt_config.json')
  15. # 静态文件目录(前端构建产物或由 nginx 提供)
  16. STATIC_FOLDER = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), 'frontend', 'dist')
  17. # 导入配置
  18. from config import (
  19. MAX_BUFFER_SIZE,
  20. FLASK_SECRET_KEY,
  21. FLASK_DEBUG,
  22. FLASK_HOST,
  23. FLASK_PORT,
  24. LOG_LEVEL,
  25. LOG_FORMAT,
  26. LOG_FILE,
  27. SOCKETIO_ASYNC_MODE,
  28. SOCKETIO_ALLOWED_ORIGINS,
  29. DEFAULT_FORWARD_SERIAL_TO_MQTT,
  30. DEFAULT_FORWARD_MQTT_TO_SERIAL,
  31. DEFAULT_MQTT_PUBLISH_TOPIC,
  32. SOCKETIO_NAMESPACE_DATA,
  33. SOCKETIO_NAMESPACE_STATUS,
  34. SOCKETIO_NAMESPACE_CONTROL,
  35. ERROR_CODES,
  36. ERROR_MESSAGES,
  37. DEFAULT_CUSTOMER_ID,
  38. DEFAULT_DTU_ID,
  39. DTU_HEARTBEAT_INTERVAL,
  40. DTU_OFFLINE_TIMEOUT,
  41. DEFAULT_MQTT_TOPIC_PREFIX,
  42. DTU_REGISTRATION_TOPIC,
  43. DTU_STATUS_TOPIC,
  44. DTU_CONTROL_TOPIC,
  45. DTU_RESPONSE_TOPIC,
  46. DTU_EVENT_TOPIC,
  47. DTU_ALARM_TOPIC,
  48. DTU_BROADCAST_TOPIC,
  49. MAX_PANELS,
  50. MAX_PORTS_PER_PANEL,
  51. DEFAULT_ADDRESS_RANGE,
  52. DEFAULT_UNBOUND_PROBE_BATCH,
  53. PANEL_MISS_THRESHOLD,
  54. PANEL_PROBE_TIMEOUT,
  55. DHT11_ENABLED,
  56. DHT11_GPIO_PIN,
  57. DHT11_GPIO_CHIP,
  58. DHT11_POLL_INTERVAL,
  59. DHT11_SIMULATE,
  60. NEXTION_ENABLED,
  61. NEXTION_SCREEN_PORT,
  62. NEXTION_SCREEN_BAUD,
  63. NEXTION_PANEL_NAME_PREFIX,
  64. NEXTION_CMD_DELAY_MS
  65. )
  66. # 配置日志
  67. logging_config = {
  68. 'level': getattr(logging, LOG_LEVEL),
  69. 'format': LOG_FORMAT
  70. }
  71. if LOG_FILE:
  72. logging_config['filename'] = LOG_FILE
  73. logging.basicConfig(**logging_config)
  74. logger = logging.getLogger('serial_mqtt_gateway')
  75. _boot_time = time.time()
  76. from modules.serial_port import SerialPort
  77. from modules.mqtt_client import MQTTClient
  78. from modules.network_config import network_manager
  79. from modules.modbus_rtu import ModbusRTUClient, ANTENNA_ADDRESSES, AddressConfigProtocol, build_confirm_address, calculate_crc16
  80. from modules.dht11_sensor import DHT11Sensor
  81. from modules.nextion_display import NextionDisplay
  82. from modules import config_backup
  83. app = Flask(__name__)
  84. app.config['SECRET_KEY'] = FLASK_SECRET_KEY
  85. # 配置CORS以允许nginx代理的前端访问
  86. CORS(app, resources={r"/api/*": {"origins": "*"}, r"/socket.io/*": {"origins": "*"}})
  87. # 初始化SocketIO
  88. socketio = SocketIO(
  89. app,
  90. cors_allowed_origins=SOCKETIO_ALLOWED_ORIGINS,
  91. async_mode=SOCKETIO_ASYNC_MODE,
  92. manage_session=False, # 禁用会话管理以提高性能
  93. ping_timeout=30, # 心跳超时时间
  94. ping_interval=25, # 心跳间隔
  95. logger=FLASK_DEBUG, # 根据Flask调试模式决定是否记录SocketIO日志
  96. engineio_logger=FLASK_DEBUG
  97. )
  98. # 初始化串口和MQTT客户端
  99. serial_client = SerialPort()
  100. mqtt_client = MQTTClient()
  101. modbus_client = ModbusRTUClient(serial_client)
  102. address_config = AddressConfigProtocol(serial_client)
  103. # Nextion 串口屏实例
  104. screen_display = NextionDisplay()
  105. screen_current_panel_index = 0
  106. screen_lock = threading.Lock()
  107. screen_refresh_queue = queue.Queue()
  108. screen_refresh_thread = None
  109. # 初始化 DHT11 传感器(后续在 __main__ 中设置回调并启动)
  110. dht11_sensor = None
  111. if DHT11_ENABLED and (DHT11_GPIO_PIN is not None or DHT11_SIMULATE):
  112. dht11_sensor = DHT11Sensor(
  113. gpio_pin=DHT11_GPIO_PIN,
  114. gpio_chip=DHT11_GPIO_CHIP,
  115. poll_interval=DHT11_POLL_INTERVAL,
  116. simulate=DHT11_SIMULATE
  117. )
  118. elif DHT11_ENABLED and DHT11_GPIO_PIN is None and not DHT11_SIMULATE:
  119. logger.warning("DHT11 已启用但未配置 GPIO pin (DHT11_GPIO_PIN),跳过本地传感器启动")
  120. # 转发标志
  121. forward_serial_to_mqtt = DEFAULT_FORWARD_SERIAL_TO_MQTT
  122. forward_mqtt_to_serial = DEFAULT_FORWARD_MQTT_TO_SERIAL
  123. mqtt_publish_topic = DEFAULT_MQTT_PUBLISH_TOPIC
  124. # DTU MQTT协议配置(运行时可修改)
  125. dtu_config = {
  126. 'topic_prefix': DEFAULT_MQTT_TOPIC_PREFIX,
  127. 'customer_id': DEFAULT_CUSTOMER_ID,
  128. 'dtu_id': DEFAULT_DTU_ID,
  129. 'firmware_version': 'v1.0.0',
  130. 'hardware_version': 'v1.0',
  131. 'heartbeat_interval': DTU_HEARTBEAT_INTERVAL,
  132. 'enabled': True,
  133. 'address_range': dict(DEFAULT_ADDRESS_RANGE)
  134. }
  135. DTU_CONFIG_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'dtu_config.json')
  136. # 本地自动快照目录(每次写配置前自动快照,可在"数据备份"页恢复)
  137. BACKUP_DIR = config_backup.BACKUP_DIR
  138. def _save_config_file(path, data, label):
  139. """快照 + 原子写 JSON 配置文件"""
  140. try:
  141. config_backup.snapshot_before_write(path, BACKUP_DIR)
  142. config_backup.write_json_atomic(path, data)
  143. return True
  144. except Exception as e:
  145. logger.error(f"保存{label}失败: {e}")
  146. return False
  147. def load_dtu_config():
  148. global dtu_config
  149. try:
  150. if os.path.exists(DTU_CONFIG_FILE):
  151. saved = None
  152. try:
  153. saved = config_backup.read_json_file(DTU_CONFIG_FILE)
  154. except Exception as e:
  155. logger.warning(f"加载DTU配置失败: {e},尝试回退快照")
  156. saved, snap = config_backup.latest_snapshot_content(DTU_CONFIG_FILE, BACKUP_DIR)
  157. if isinstance(saved, dict):
  158. logger.warning(f"已从快照恢复DTU配置: {snap}")
  159. if isinstance(saved, dict):
  160. dtu_config.update(saved)
  161. except Exception as e:
  162. logger.warning(f"加载DTU配置失败: {e}")
  163. def save_dtu_config():
  164. _save_config_file(DTU_CONFIG_FILE, dtu_config, 'DTU配置')
  165. # 端口状态追踪(用于事件检测)
  166. port_state = {} # {panel_id: {port_id: {'last_uid': str, 'expected_uid': str, 'alarm_count': int}}}
  167. # jumper_uid 十六进制字符集(大写)
  168. _JUMPER_HEX = set('0123456789ABCDEF')
  169. _mac_uid_prefix_cache = None # 缓存 MAC 派生的前缀(稳定,避免每次发布都读 MAC)
  170. def _read_mac_bytes():
  171. """读取本机真实 MAC 地址(6 字节),失败返回 None。
  172. 优先用 uuid.getnode();其 multicast 位为 1 时是随机生成而非真实硬件 MAC,
  173. 改读 /sys/class/net(Linux)兜底。
  174. """
  175. try:
  176. node = uuid.getnode()
  177. # 第 9 位(0x010000000000)为 1 表示 multicast/随机 MAC,非真实硬件地址
  178. if node and not (node & 0x010000000000):
  179. return node.to_bytes(6, 'big')
  180. except Exception:
  181. pass
  182. try:
  183. net_dir = '/sys/class/net'
  184. if os.path.isdir(net_dir):
  185. for iface in sorted(os.listdir(net_dir)):
  186. if iface == 'lo':
  187. continue
  188. path = os.path.join(net_dir, iface, 'address')
  189. if os.path.exists(path):
  190. with open(path) as f:
  191. mac_str = f.read().strip()
  192. if re.match(r'^[0-9A-Fa-f]{2}(:[0-9A-Fa-f]{2}){5}$', mac_str):
  193. return bytes(int(b, 16) for b in mac_str.split(':'))
  194. except Exception:
  195. pass
  196. return None
  197. def _dtu_uid_prefix():
  198. """jumper_uid 前 4 位 DTU 号(4 位大写十六进制),从 MAC 派生保证多 DTU 唯一。
  199. 适配规则: MQTT jumper_uid = 前 4 位 DTU 号 + 后 12 位硬件卡号(共 16 位)。
  200. 优先级:
  201. 1. dtu_config['uid_prefix'] 显式配置(4 位大写十六进制)
  202. 2. 本机 MAC 地址的 CRC16(稳定、各 DTU 不同)
  203. 3. 兜底: dtu_id 末尾数字
  204. """
  205. global _mac_uid_prefix_cache
  206. p = (dtu_config.get('uid_prefix') or '').strip().upper()
  207. if len(p) == 4 and all(c in _JUMPER_HEX for c in p):
  208. return p
  209. if _mac_uid_prefix_cache is None:
  210. mac = _read_mac_bytes()
  211. if mac:
  212. _mac_uid_prefix_cache = calculate_crc16(mac).hex().upper()
  213. else:
  214. m = re.search(r'(\d+)$', str(dtu_config.get('dtu_id') or ''))
  215. n = int(m.group(1)) if m else 0
  216. _mac_uid_prefix_cache = f'{n & 0xFFFF:04X}'
  217. return _mac_uid_prefix_cache
  218. def _to_mqtt_jumper_uid(card_uid):
  219. """内部 12 位硬件卡号 -> MQTT 16 位 jumper_uid(前 4 位 DTU 号 + 后 12 位卡号)。
  220. None/空 -> None。硬件通信固定 6 字节(12 位),仅在 MQTT 边界拼成 16 位。
  221. """
  222. if not card_uid:
  223. return None
  224. return _dtu_uid_prefix() + card_uid.upper()
  225. def _from_mqtt_jumper_uid(value):
  226. """MQTT 16 位 jumper_uid -> 内部 12 位硬件卡号(剥前 4 位 DTU 号)。
  227. None/空 -> None;非 16 位大写十六进制 -> 抛 ValueError(由 SYNC 调用方拒绝)。
  228. """
  229. if value is None:
  230. return None
  231. if not isinstance(value, str):
  232. raise ValueError(f"jumper_uid 必须是字符串或 null,当前类型: {type(value).__name__}")
  233. v = value.strip().upper()
  234. if v == '':
  235. return None
  236. if len(v) != 16 or any(c not in _JUMPER_HEX for c in v):
  237. raise ValueError(f"jumper_uid 必须是 16 位大写十六进制或 null,当前: {value!r}")
  238. return v[4:]
  239. def _to_internal_uid_lenient(value):
  240. """重建端口状态用:兼容历史持久化数据。
  241. 16 位 -> 剥前 4 位得 12 位;12 位 -> 原样;None/空 -> None;非法 -> None。
  242. """
  243. if value is None:
  244. return None
  245. if not isinstance(value, str):
  246. return None
  247. v = value.strip().upper()
  248. if v == '':
  249. return None
  250. if len(v) == 16 and all(c in _JUMPER_HEX for c in v):
  251. return v[4:]
  252. if len(v) == 12 and all(c in _JUMPER_HEX for c in v):
  253. return v
  254. return None
  255. def _build_card_mappings_for_panel(panel_id):
  256. """从 dtu_config port_mappings 构建该面板 24 路卡号映射,用于 V4 下发硬件。
  257. 返回: {port_id(1-24): 'hex12' or None}
  258. 未配置/非法的端口返回 None(set_card_mapping 内部填 0xFF*6 无卡标记)。
  259. """
  260. mappings = dtu_config.get('port_mappings', {}).get(panel_id, {})
  261. result = {}
  262. for port_id in range(1, 25):
  263. uid = mappings.get(str(port_id))
  264. result[port_id] = _to_internal_uid_lenient(uid) # 非法/空 -> None
  265. return result
  266. def get_sorted_panels():
  267. """按 position 排序返回 panel 列表。"""
  268. return sorted(panel_config.items(), key=lambda x: x[1].get('position', 0))
  269. def _port_status_to_pic(port_state_entry):
  270. """将端口状态推导为 Nextion pic 编号。
  271. pic 映射(当前屏幕工程只有 pic 0-3,ILLEGAL_DISCONNECT 复用 pic=2 告警图):
  272. 0 = UNKNOWN (未轮询)
  273. 1 = DISCONNECTED (正常空闲: 无期望且无卡)
  274. 2 = 告警 (ILLEGAL_CONNECT 非法插入 + ILLEGAL_DISCONNECT 非法拔出)
  275. 3 = CONNECTED (正常已连: 实际 == 期望)
  276. 未来若 Nextion 工程增加 pic=4,将 ILLEGAL_DISCONNECT 单独改为 return 4。
  277. """
  278. if not port_state_entry:
  279. return 0
  280. last_uid = port_state_entry.get('last_uid')
  281. expected_uid = port_state_entry.get('expected_uid')
  282. last_polled = port_state_entry.get('last_polled_at')
  283. if not last_polled:
  284. return 0 # UNKNOWN
  285. if last_uid is None:
  286. return 2 if expected_uid else 1 # ILLEGAL_DISCONNECT 复用告警图 / DISCONNECTED
  287. if expected_uid and last_uid != expected_uid:
  288. return 2 # ILLEGAL_CONNECT
  289. return 3 # CONNECTED
  290. def _get_display_ip():
  291. """获取用于屏幕显示的本机 IP 地址。
  292. 优先选择 192.168.199.x 管理网段;其次选择物理网卡 eth*/en* 上的 IPv4;
  293. 最后回退到任意非回环 IPv4 或静态配置 IP。
  294. """
  295. try:
  296. net_status = network_manager.get_network_status()
  297. interfaces = net_status.get('interfaces', {})
  298. # 1. 优先 192.168.199.x 管理网段
  299. for iface, data in interfaces.items():
  300. if iface in ('lo', 'docker0') or iface.startswith(('br-', 'veth')):
  301. continue
  302. for addr in data.get('ip_addresses', []):
  303. if addr.startswith('192.168.199.') and '.' in addr:
  304. return addr
  305. # 2. 优先物理网卡 eth*/en*/wlan* 上的 IPv4
  306. physical_ips = []
  307. for iface, data in interfaces.items():
  308. if iface in ('lo', 'docker0') or iface.startswith(('br-', 'veth', 'docker')):
  309. continue
  310. if not (iface.startswith(('eth', 'en', 'wlan', 'wl')) or iface == 'end0'):
  311. continue
  312. for addr in data.get('ip_addresses', []):
  313. if '.' in addr and not addr.startswith('127.'):
  314. physical_ips.append((iface, addr))
  315. # 优先有线接口
  316. for iface, addr in physical_ips:
  317. if iface.startswith(('eth', 'en', 'end')):
  318. return addr
  319. if physical_ips:
  320. return physical_ips[0][1]
  321. # 3. 任意非回环 IPv4
  322. for iface, data in interfaces.items():
  323. if iface in ('lo', 'docker0') or iface.startswith(('br-', 'veth')):
  324. continue
  325. for addr in data.get('ip_addresses', []):
  326. if '.' in addr and not addr.startswith('127.'):
  327. return addr
  328. # 4. 回退:尝试网络配置中的静态 IP
  329. net_config = network_manager.get_network_config()
  330. static_ip = net_config.get('static_config', {}).get('ip_address')
  331. if static_ip:
  332. return static_ip
  333. except Exception as e:
  334. logger.warning(f"获取显示 IP 失败: {e}")
  335. return ''
  336. _last_display_ip = None # 上次写入屏幕的 IP,用于检测变化
  337. def _network_ip_monitor_loop():
  338. """后台监测本机 IP 变化。
  339. 插网线获取到 IP(或 IP 发生任何变化)时,及时刷新串口屏显示。
  340. 复用 screen_refresh_queue 串行化下发,避免与其它刷新竞争。
  341. """
  342. global _last_display_ip
  343. while True:
  344. try:
  345. current_ip = _get_display_ip()
  346. if current_ip != _last_display_ip:
  347. logger.info(f"显示 IP 变化: {_last_display_ip!r} -> {current_ip!r}")
  348. _last_display_ip = current_ip
  349. if screen_display.get_status().get('connected'):
  350. panels = get_sorted_panels()
  351. if panels:
  352. with screen_lock:
  353. idx = screen_current_panel_index
  354. if 0 <= idx < len(panels):
  355. screen_refresh_queue.put((panels[idx][0],))
  356. else:
  357. # 没有配线架也要刷新 IP,避免屏幕停留在 HMI 默认文字
  358. screen_refresh_queue.put((None,))
  359. except Exception as e:
  360. logger.error(f"网络 IP 监测异常: {e}")
  361. time.sleep(3)
  362. def _screen_refresh_worker():
  363. """在后台线程中异步刷新 Nextion 屏幕。"""
  364. while True:
  365. try:
  366. item = screen_refresh_queue.get()
  367. if item is None:
  368. break
  369. panel_id = item[0]
  370. if panel_id is None:
  371. # 没有配线架时也要把 IP/状态刷到屏幕
  372. if screen_display.get_status().get('connected'):
  373. _send_screen_header()
  374. else:
  375. refresh_screen_panel(panel_id)
  376. except Exception as e:
  377. logger.error(f"屏幕刷新工作线程异常: {e}")
  378. finally:
  379. try:
  380. screen_refresh_queue.task_done()
  381. except Exception:
  382. pass
  383. def _send_screen_header(panel_name=None):
  384. """下发与具体 panel 无关的屏幕信息:本机 IP、终端名、版本、MQTT 状态。
  385. 即使一个配线架都没发现(panel_count=0)也调用,否则屏幕会一直停留在
  386. HMI 工程的默认文字(例如默认占位 IP)。
  387. """
  388. online_count = sum(
  389. 1 for pid, _ in get_sorted_panels()
  390. if device_last_seen.get(pid, 0) > time.time() - 60
  391. )
  392. mqtt_st = mqtt_client.get_status()
  393. mqtt_connected = mqtt_st.get('connected', False) if isinstance(mqtt_st, dict) else bool(mqtt_st)
  394. delay_ms = NEXTION_CMD_DELAY_MS
  395. if panel_name is None:
  396. panel_name = f'{NEXTION_PANEL_NAME_PREFIX}0'
  397. screen_display.send_cmd(f't2.txt="{panel_name}"', delay_ms=delay_ms)
  398. screen_display.send_cmd(f't1.txt="{_get_display_ip()}"', delay_ms=delay_ms)
  399. screen_display.send_cmd(f't4.txt="{dtu_config.get("firmware_version", "v1.0.0")}"', delay_ms=delay_ms)
  400. mqtt_str = '已连接' if mqtt_connected else '未连接'
  401. screen_display.send_cmd(f'g0.txt="MQTT:{mqtt_str} 当前终端数:{online_count}"', delay_ms=delay_ms)
  402. screen_display.send_cmd(f'g0.pco={0 if mqtt_connected else 63488}', delay_ms=delay_ms)
  403. def refresh_screen_panel(panel_id):
  404. """将指定 panel 的数据下发到 Nextion 屏幕。"""
  405. global screen_current_panel_index
  406. with screen_lock:
  407. panels = get_sorted_panels()
  408. if not panels:
  409. return False, '没有可用的 panel'
  410. panel_ids = [p[0] for p in panels]
  411. if panel_id not in panel_ids:
  412. return False, f'panel 不存在: {panel_id}'
  413. if not screen_display.get_status().get('connected'):
  414. return False, '屏幕串口未连接'
  415. idx = panel_ids.index(panel_id)
  416. screen_current_panel_index = idx
  417. # 当前 panel 对应的设备地址
  418. device_address = panel_config.get(panel_id, {}).get('address', 1)
  419. _send_screen_header(panel_name=f'{NEXTION_PANEL_NAME_PREFIX}{idx + 1}')
  420. for port_id in range(1, 25):
  421. # 屏幕网口图片显示 LED:由端口真实连接状态推导(连接/断开/非法/未知)
  422. pic = _port_status_to_pic(port_state.get(panel_id, {}).get(port_id))
  423. cmd = f'p{port_id - 1}.pic={pic}'
  424. ok = screen_display.send_cmd(cmd, delay_ms=delay_ms)
  425. logger.info(f"[Nextion] panel={panel_id} addr={device_address} port={port_id} pic={pic} cmd={cmd} ok={ok}")
  426. logger.info(f"屏幕已刷新: {panel_id} (终端{idx + 1})")
  427. return True, '屏幕已刷新'
  428. def switch_screen_panel(direction):
  429. """切换当前显示的 panel,跳过已标记离线的终端。
  430. 离线终端不再占用屏幕:按"上一/下一终端"时自动落到下一台在线的设备,
  431. 避免屏上长时间停留在空白面板上。若全部终端都离线,则退化为相邻切换,
  432. 方便用户仍能浏览各终端的离线状态。
  433. """
  434. global screen_current_panel_index
  435. with screen_lock:
  436. panels = get_sorted_panels()
  437. if not panels:
  438. return False, '没有可用的 panel'
  439. n = len(panels)
  440. start = screen_current_panel_index
  441. idx = (start + direction) % n
  442. found = False
  443. while idx != start:
  444. if panels[idx][1].get('status') != 'offline':
  445. found = True
  446. break
  447. idx = (idx + direction) % n
  448. if not found:
  449. # 全部离线:仍然切换到相邻面板,让用户能浏览各终端
  450. idx = (start + direction) % n
  451. screen_current_panel_index = idx
  452. panel_id = panels[idx][0]
  453. logger.info(f'屏幕切换: 终端{idx + 1} -> {panel_id}')
  454. return refresh_screen_panel(panel_id)
  455. def _refresh_current_screen_panel():
  456. """LED 等状态变化后,触发当前屏幕 panel 刷新。"""
  457. try:
  458. panels = get_sorted_panels()
  459. if not panels:
  460. return
  461. with screen_lock:
  462. idx = screen_current_panel_index
  463. if 0 <= idx < len(panels):
  464. panel_id = panels[idx][0]
  465. screen_refresh_queue.put((panel_id,))
  466. except Exception as e:
  467. logger.warning(f"触发屏幕刷新失败: {e}")
  468. def screen_button_event_handler(page_id, comp_id, event):
  469. """处理屏幕按钮事件。"""
  470. if event != 0x01:
  471. return
  472. if page_id == 0x00 and comp_id == 0x05:
  473. logger.info('串口屏事件: 上一终端')
  474. switch_screen_panel(-1)
  475. elif page_id == 0x00 and comp_id == 0x06:
  476. logger.info('串口屏事件: 下一终端')
  477. switch_screen_panel(1)
  478. # 面板配置(从地址配置模块加载)
  479. panel_config = {} # {panel_id: {'address': int, 'position': int, 'panel_uid': str}}
  480. # 面板连续轮询失败计数(达到阈值后标记离线,并在后续轮次改用 uid=0 探测:换架接管/原架恢复)
  481. panel_fail_count = {} # {panel_id: int}
  482. # 未绑定地址轮转探测游标(地址段扫描用,避免每轮全扫空地址)
  483. _unbound_probe_cursor = None
  484. # 手动"扫描新线架"后台任务状态
  485. _panel_scan_state = {'running': False, 'found': [], 'started_at': None, 'finished_at': None, 'error': None}
  486. def _address_range():
  487. """当前地址段 (start, end),非法时回退默认值"""
  488. rng = dtu_config.get('address_range') or DEFAULT_ADDRESS_RANGE
  489. try:
  490. start = int(rng.get('start', DEFAULT_ADDRESS_RANGE['start']))
  491. end = int(rng.get('end', DEFAULT_ADDRESS_RANGE['end']))
  492. except Exception:
  493. start, end = DEFAULT_ADDRESS_RANGE['start'], DEFAULT_ADDRESS_RANGE['end']
  494. if not (1 <= start <= end <= 247):
  495. start, end = DEFAULT_ADDRESS_RANGE['start'], DEFAULT_ADDRESS_RANGE['end']
  496. return start, end
  497. def _default_panel_id(addr):
  498. return f"PANEL_{dtu_config.get('dtu_id', 'DTU001')}_{addr}"
  499. def _binding_uid_for_address(addr):
  500. """返回绑定到该地址的线架 UID(无绑定返回 None)"""
  501. for uid, a in address_config.get_stored_devices().items():
  502. if a == addr:
  503. return uid
  504. return None
  505. def _panel_id_for_address(addr):
  506. """按地址找 panel_id(panel_config 优先,其次 panel_id_mapping,最后默认命名)"""
  507. for pid, cfg in panel_config.items():
  508. if cfg.get('address') == addr:
  509. return pid
  510. pm = dtu_config.get('panel_id_mapping', {}) or {}
  511. uid = _binding_uid_for_address(addr)
  512. custom = (pm.get(uid) if uid else None) or pm.get(addr)
  513. return str(custom) if custom else _default_panel_id(addr)
  514. def _is_panel_custom_id(panel_id, addr):
  515. """panel_id 是否为 panel_id_mapping 自定义(区别于默认命名)"""
  516. return panel_id != _default_panel_id(addr)
  517. def _ensure_panel_entry(panel_id, addr, uid, status='online'):
  518. panel_config[panel_id] = {
  519. 'address': addr,
  520. 'position': addr,
  521. 'panel_uid': uid,
  522. 'status': status,
  523. }
  524. def _mark_panel_offline_once(panel_id, addr, cfg):
  525. """标记面板离线(只发布一次离线状态,清端口运行态)"""
  526. if cfg.get('status') == 'offline':
  527. return
  528. cfg['status'] = 'offline'
  529. port_state.pop(panel_id, None)
  530. logger.info(f"面板 {panel_id}(地址={addr}) 已标记离线")
  531. dtu_publish_panel_status(panel_id, addr, None, online=False)
  532. # 若该面板正显示在串口屏上,触发刷新使其显示离线(全灰)状态,避免残留旧画面
  533. _refresh_current_screen_panel()
  534. def _handle_panel_read_failure(panel_id, addr, cfg):
  535. """面板读卡失败:累计 miss;达到阈值标记离线。
  536. 返回 True 表示已进入"接管/恢复探测"阶段(后续轮次用 uid=0 探测该地址)。
  537. """
  538. panel_fail_count[panel_id] = panel_fail_count.get(panel_id, 0) + 1
  539. n = panel_fail_count[panel_id]
  540. if n < PANEL_MISS_THRESHOLD:
  541. logger.warning(f"面板 {panel_id}(地址={addr}) 轮询无响应,连续失败 {n} 次")
  542. return False
  543. if cfg.get('status') != 'offline':
  544. logger.warning(f"面板 {panel_id}(地址={addr}) 连续 {n} 次无响应,转入接管/恢复探测")
  545. _mark_panel_offline_once(panel_id, addr, cfg)
  546. return True
  547. def _maybe_refresh_screen_panel(panel_id):
  548. """该面板若正显示在串口屏上,触发一次刷新"""
  549. try:
  550. panels = get_sorted_panels()
  551. trigger_panel_id = None
  552. with screen_lock:
  553. if panels and 0 <= screen_current_panel_index < len(panels) and panels[screen_current_panel_index][0] == panel_id:
  554. trigger_panel_id = panel_id
  555. if trigger_panel_id:
  556. screen_refresh_queue.put((trigger_panel_id,))
  557. except Exception as e:
  558. logger.error(f"刷新屏幕失败: {e}")
  559. def _migrate_port_mappings(old_panel_id, new_panel_id):
  560. """线架地址变化时迁移端口期望映射(随 panel_id 一起搬到新地址下)"""
  561. pm = dtu_config.setdefault('port_mappings', {})
  562. if old_panel_id and old_panel_id != new_panel_id and old_panel_id in pm and new_panel_id not in pm:
  563. pm[new_panel_id] = pm.pop(old_panel_id)
  564. save_dtu_config()
  565. logger.info(f"端口映射已迁移: {old_panel_id} -> {new_panel_id}")
  566. def _push_card_mappings_to_panel(panel_id, addr, reason=''):
  567. """把持久化的期望卡号下发到线架硬件(换架接管/新注册后恢复面板数据)"""
  568. try:
  569. mappings = _build_card_mappings_for_panel(panel_id)
  570. if not any(v for v in mappings.values()):
  571. return
  572. result = modbus_client.set_card_mapping(addr, mappings)
  573. if 'error' in result:
  574. logger.warning(f"{reason}下发卡号失败: 地址={addr}, {result.get('error')}")
  575. else:
  576. count = len([v for v in mappings.values() if v])
  577. logger.info(f"{reason}已向地址={addr} 下发 {count} 路卡号")
  578. except Exception as e:
  579. logger.error(f"{reason}下发卡号异常: 地址={addr}, {e}")
  580. def _process_cards_result(panel_id, addr, result, silent=False):
  581. """处理一次成功的批量读卡结果:更新端口状态/事件/告警,返回 panel_ports(用于状态发布)
  582. silent=True 时只对齐端口状态(不发布事件/告警),用于换架接管后的静默重建。
  583. """
  584. panel_ports = []
  585. for card in result.get('cards', []):
  586. port_id = card['antenna']
  587. uid = card['card_str'].upper() if card.get('present') else ''
  588. if panel_id not in port_state:
  589. port_state[panel_id] = {}
  590. if port_id not in port_state[panel_id]:
  591. # 重建端口状态时,从持久化的期望映射回填 expected_uid(内部 12 位)。
  592. # 兼容历史脏数据:16 位剥前 4 位得 12 位;12 位原样;非法丢弃并告警。
  593. exp_uid = dtu_config.get('port_mappings', {}).get(panel_id, {}).get(str(port_id))
  594. norm = _to_internal_uid_lenient(exp_uid)
  595. if exp_uid and norm is None:
  596. logger.warning(f"面板{panel_id}端口{port_id} 持久化的 expected_uid 非法已丢弃: {exp_uid!r}")
  597. port_state[panel_id][port_id] = {
  598. 'last_uid': None,
  599. 'expected_uid': norm,
  600. 'alarm_count': 0,
  601. 'last_polled_at': int(time.time() * 1000)
  602. }
  603. ps = port_state[panel_id][port_id]
  604. ps['last_polled_at'] = int(time.time() * 1000)
  605. last_uid = ps.get('last_uid')
  606. if silent:
  607. ps['last_uid'] = uid or None
  608. ps['alarm_count'] = 0
  609. else:
  610. if uid and uid != last_uid:
  611. dtu_publish_event(panel_id, port_id, 'CONNECT' if not last_uid else 'MOVE', uid, last_uid)
  612. ps['last_uid'] = uid
  613. elif not uid and last_uid:
  614. dtu_publish_event(panel_id, port_id, 'DISCONNECT', None, last_uid)
  615. ps['last_uid'] = None
  616. ps_exp = ps.get('expected_uid')
  617. if dtu_config.get('alarm_enabled', True):
  618. if ps_exp and uid and uid != ps_exp:
  619. ps['alarm_count'] = ps.get('alarm_count', 0) + 1
  620. sev = 'CRITICAL' if ps['alarm_count'] >= 3 else 'WARNING'
  621. dtu_publish_alarm(panel_id, port_id, 'ILLEGAL_CONNECT', ps_exp, uid, sev)
  622. elif ps_exp and not uid:
  623. ps['alarm_count'] = ps.get('alarm_count', 0) + 1
  624. sev = 'CRITICAL' if ps['alarm_count'] >= 3 else 'WARNING'
  625. dtu_publish_alarm(panel_id, port_id, 'ILLEGAL_DISCONNECT', ps_exp, None, sev)
  626. elif ps_exp and uid == ps_exp:
  627. ps['alarm_count'] = 0
  628. ps_exp = ps.get('expected_uid')
  629. if uid:
  630. port_status = 'ILLEGAL' if (ps_exp and uid != ps_exp) else 'CONNECTED'
  631. else:
  632. port_status = 'DISCONNECTED'
  633. panel_ports.append({'port_id': port_id, 'status': port_status, 'jumper_uid': uid or None})
  634. # V4: LED 由配线架自控,DTU 不再自动下发灯状态
  635. return panel_ports
  636. def _handle_panel_read_success(panel_id, cfg, addr, result, silent=False):
  637. """一次成功读卡后的统一处理:端口状态 + 恢复在线 + 状态发布 + 屏幕刷新"""
  638. panel_ports = _process_cards_result(panel_id, addr, result, silent=silent)
  639. if not silent and (cfg.get('status') == 'offline' or panel_fail_count.get(panel_id, 0) > 0):
  640. cfg['status'] = 'online'
  641. panel_fail_count.pop(panel_id, None)
  642. logger.info(f"面板 {panel_id}(地址={addr}) 恢复在线")
  643. device_last_seen[panel_id] = time.time()
  644. if cfg.get('panel_uid'):
  645. device_last_seen[cfg['panel_uid']] = time.time()
  646. dtu_publish_panel_status(panel_id, addr, panel_ports)
  647. _maybe_refresh_screen_panel(panel_id)
  648. def _takeover_panel(panel_id, addr, old_uid, new_uid, result):
  649. """换架自动接管:新线架(UID 不同)出现在原地址 -> 绑定新 UID,继承原地址/映射/卡号数据"""
  650. old_uid = (old_uid or '').lower()
  651. new_uid = (new_uid or '').lower()
  652. stored = address_config.stored_devices
  653. if old_uid:
  654. stored.pop(old_uid, None)
  655. stored[new_uid] = addr
  656. save_device_config()
  657. cfg = panel_config.get(panel_id)
  658. if cfg is None:
  659. _ensure_panel_entry(panel_id, addr, new_uid)
  660. cfg = panel_config[panel_id]
  661. # 自定义 panel_id:换 UID 后仍保留原 panel_id(补一条 uid 映射,防止命名回退)
  662. if _is_panel_custom_id(panel_id, addr):
  663. dtu_config.setdefault('panel_id_mapping', {})[new_uid] = panel_id
  664. save_dtu_config()
  665. cfg['panel_uid'] = new_uid
  666. cfg['status'] = 'online'
  667. panel_fail_count.pop(panel_id, None)
  668. logger.info(f"线架自动接管: 地址={addr} 旧UID={old_uid or '未记录'} -> 新UID={new_uid},继承原面板数据(端口映射不变)")
  669. # 端口状态静默对齐(避免换架瞬间误报 CONNECT/MOVE 事件),再按期望映射下发卡号
  670. _handle_panel_read_success(panel_id, cfg, addr, result, silent=True)
  671. _push_card_mappings_to_panel(panel_id, addr, reason='换架接管后')
  672. try:
  673. dtu_register()
  674. except Exception as e:
  675. logger.warning(f"接管后注册上报失败: {e}")
  676. return True
  677. def _register_new_panel(addr, uid, result):
  678. """未绑定地址探测到线架 -> 绑定 地址↔UID"""
  679. uid = (uid or '').lower()
  680. existing_addr = address_config.stored_devices.get(uid)
  681. if existing_addr is not None and existing_addr != addr:
  682. old_pid = _panel_id_for_address(existing_addr)
  683. old_cfg = panel_config.get(old_pid, {})
  684. if old_cfg.get('status') == 'offline' or panel_fail_count.get(old_pid, 0) > 0:
  685. # 线架被手动改址:迁移绑定并把端口映射数据带到新地址
  686. address_config.stored_devices.pop(uid, None)
  687. panel_config.pop(old_pid, None)
  688. port_state.pop(old_pid, None)
  689. panel_fail_count.pop(old_pid, None)
  690. _migrate_port_mappings(old_pid, _panel_id_for_address(addr))
  691. logger.info(f"线架地址变更: UID={uid} {existing_addr} -> {addr},绑定与映射已迁移")
  692. else:
  693. logger.warning(f"地址 {addr} 探测到线架 {uid},但其仍绑定在地址 {existing_addr} 且在线,忽略本次绑定")
  694. return False
  695. address_config.stored_devices[uid] = addr
  696. save_device_config()
  697. panel_id = _panel_id_for_address(addr)
  698. if panel_id not in panel_config:
  699. _ensure_panel_entry(panel_id, addr, uid)
  700. logger.info(f"线架注册: 地址={addr} UID={uid}")
  701. _handle_panel_read_success(panel_id, panel_config[panel_id], addr, result)
  702. # 该地址若已有期望卡号(历史数据/恢复备份),一并下发到新线架
  703. _push_card_mappings_to_panel(panel_id, addr, reason='新线架注册后')
  704. try:
  705. dtu_register()
  706. except Exception as e:
  707. logger.warning(f"新线架注册后上报失败: {e}")
  708. return True
  709. def _sweep_panel_addresses():
  710. """一轮地址段顺序扫描(注册模型核心):
  711. - 已绑定且正常:带绑定 UID 读卡(读卡帧内含 UID,兼作线架保活)
  712. - 已绑定且失败过:改用 uid=0 探测 -> 同 UID=恢复在线;不同 UID=换架自动接管;无响应=继续缺失
  713. - 未绑定地址:仅当无面板处于缺失/接管状态时轮转抽查(uid=0 探测),
  714. 未配置线架收到探测帧会采纳该地址并回的自身 UID,DTU 据此绑定
  715. 所有总线帧仍走 serial_client 发送队列。
  716. """
  717. global _unbound_probe_cursor
  718. start, end = _address_range()
  719. stored = address_config.get_stored_devices()
  720. bound_map = {}
  721. for uid, a in stored.items():
  722. bound_map.setdefault(a, uid)
  723. # 任一绑定处于失败/缺失状态时暂停未绑定地址探测,避免新线架抢占非目标地址
  724. any_missing = any(panel_fail_count.get(_panel_id_for_address(a), 0) > 0 for a in bound_map)
  725. probe_set = set()
  726. if not any_missing:
  727. unbound = [a for a in range(start, end + 1) if a not in bound_map]
  728. if unbound:
  729. batch = dtu_config.get('unbound_probe_batch', DEFAULT_UNBOUND_PROBE_BATCH)
  730. try:
  731. batch = max(1, int(batch))
  732. except Exception:
  733. batch = DEFAULT_UNBOUND_PROBE_BATCH
  734. if _unbound_probe_cursor is None or not (start <= _unbound_probe_cursor <= end):
  735. _unbound_probe_cursor = start
  736. ordered = ([a for a in unbound if a >= _unbound_probe_cursor]
  737. + [a for a in unbound if a < _unbound_probe_cursor])
  738. probe_set = set(ordered[:batch])
  739. _unbound_probe_cursor = (max(probe_set) + 1) if probe_set else start
  740. if _unbound_probe_cursor > end:
  741. _unbound_probe_cursor = start
  742. sweep_addrs = sorted(set(range(start, end + 1)) | set(bound_map.keys()))
  743. online = 0
  744. offline = 0
  745. for addr in sweep_addrs:
  746. uid = bound_map.get(addr)
  747. try:
  748. if not uid:
  749. # 未绑定地址:轮转抽查
  750. if addr in probe_set:
  751. result = modbus_client.read_all_antenna_cards(addr, uid=None, timeout=PANEL_PROBE_TIMEOUT)
  752. if 'error' not in result:
  753. new_uid = (result.get('panel_uid') or '').lower()
  754. if not new_uid:
  755. logger.warning(f"地址 {addr} 探测有响应但未携带线架 UID(旧固件?),忽略绑定")
  756. else:
  757. _register_new_panel(addr, new_uid, result)
  758. continue
  759. panel_id = _panel_id_for_address(addr)
  760. cfg = panel_config.get(panel_id)
  761. if cfg is None:
  762. _ensure_panel_entry(panel_id, addr, uid)
  763. cfg = panel_config[panel_id]
  764. if panel_fail_count.get(panel_id, 0) == 0:
  765. # 正常轮次:带绑定 UID 读卡
  766. try:
  767. uid_bytes = bytes.fromhex(uid)
  768. except ValueError:
  769. uid_bytes = None
  770. if uid_bytes is None:
  771. logger.error(f"地址 {addr} 绑定的 UID 非法: {uid!r},跳过本轮")
  772. continue
  773. result = modbus_client.read_all_antenna_cards(addr, uid=uid_bytes, timeout=1.0)
  774. if 'error' in result:
  775. offline += 1
  776. _handle_panel_read_failure(panel_id, addr, cfg)
  777. else:
  778. online += 1
  779. _handle_panel_read_success(panel_id, cfg, addr, result)
  780. else:
  781. # 缺失/接管探测轮次:uid=0 探测(原线架恢复或新线架换架接管)
  782. result = modbus_client.read_all_antenna_cards(addr, uid=None, timeout=PANEL_PROBE_TIMEOUT)
  783. if 'error' in result:
  784. offline += 1
  785. _handle_panel_read_failure(panel_id, addr, cfg)
  786. continue
  787. resp_uid = (result.get('panel_uid') or '').lower()
  788. if not resp_uid:
  789. # 旧固件无 UID 字段:无法校验,按原 UID 视作在线(过渡期)
  790. logger.warning(f"地址 {addr} 探测响应无 UID(旧固件?),按原绑定 UID 视作在线")
  791. online += 1
  792. _handle_panel_read_success(panel_id, cfg, addr, result)
  793. elif resp_uid == uid.lower():
  794. online += 1
  795. _handle_panel_read_success(panel_id, cfg, addr, result)
  796. else:
  797. _takeover_panel(panel_id, addr, uid, resp_uid, result)
  798. online += 1
  799. finally:
  800. time.sleep(0.05)
  801. return online, offline
  802. # 数据存储缓冲区
  803. serial_data_buffer = []
  804. mqtt_data_buffer = []
  805. # 状态标志
  806. serial_status = False
  807. mqtt_status = False
  808. # 辅助函数:统一处理 get_status() 返回 dict 的布尔判断
  809. def _mqtt_connected():
  810. """MQTT 客户端是否已连接"""
  811. st = mqtt_client.get_status()
  812. return st.get('connected', False) if isinstance(st, dict) else bool(st)
  813. def _serial_connected():
  814. """串口是否已连接"""
  815. st = serial_client.get_status()
  816. return st.get('connected', False) if isinstance(st, dict) else bool(st)
  817. # 串口配置保存和加载函数
  818. def save_serial_config(port, baudrate=9600, timeout=1.0, bytesize=8, parity='N', stopbits=1):
  819. """保存串口配置"""
  820. config = {
  821. 'port': port,
  822. 'baudrate': baudrate,
  823. 'timeout': timeout,
  824. 'bytesize': bytesize,
  825. 'parity': parity,
  826. 'stopbits': stopbits
  827. }
  828. if _save_config_file(SERIAL_CONFIG_FILE, config, '串口配置'):
  829. logger.info(f"串口配置已保存: {port} @ {baudrate}")
  830. def load_serial_config():
  831. """加载串口配置(文件损坏时回退最新快照)"""
  832. if os.path.exists(SERIAL_CONFIG_FILE):
  833. try:
  834. try:
  835. config = config_backup.read_json_file(SERIAL_CONFIG_FILE)
  836. except Exception as e:
  837. logger.error(f"加载串口配置失败: {e},尝试回退快照")
  838. config, snap = config_backup.latest_snapshot_content(SERIAL_CONFIG_FILE, BACKUP_DIR)
  839. if not isinstance(config, dict):
  840. return None
  841. logger.warning(f"已从快照恢复串口配置: {snap}")
  842. logger.info(f"已加载串口配置: {config.get('port')} @ {config.get('baudrate')}")
  843. return config
  844. except Exception as e:
  845. logger.error(f"加载串口配置失败: {e}")
  846. return None
  847. def save_mqtt_config(broker, port=1883, client_id='', username='', password='', keepalive=60, topics=None):
  848. """保存MQTT配置"""
  849. config = {
  850. 'broker': broker,
  851. 'port': port,
  852. 'client_id': client_id,
  853. 'username': username,
  854. 'password': password,
  855. 'keepalive': keepalive,
  856. 'topics': topics or []
  857. }
  858. if _save_config_file(MQTT_CONFIG_FILE, config, 'MQTT配置'):
  859. logger.info(f"MQTT配置已保存: {broker}:{port}")
  860. def load_mqtt_config():
  861. """加载MQTT配置(文件损坏时回退最新快照)"""
  862. if os.path.exists(MQTT_CONFIG_FILE):
  863. try:
  864. try:
  865. config = config_backup.read_json_file(MQTT_CONFIG_FILE)
  866. except Exception as e:
  867. logger.error(f"加载MQTT配置失败: {e},尝试回退快照")
  868. config, snap = config_backup.latest_snapshot_content(MQTT_CONFIG_FILE, BACKUP_DIR)
  869. if not isinstance(config, dict):
  870. return None
  871. logger.warning(f"已从快照恢复MQTT配置: {snap}")
  872. logger.info(f"已加载MQTT配置: {config.get('broker')}:{config.get('port')}")
  873. return config
  874. except Exception as e:
  875. logger.error(f"加载MQTT配置失败: {e}")
  876. return None
  877. def auto_connect_mqtt():
  878. """自动连接上次使用的MQTT服务器"""
  879. config = load_mqtt_config()
  880. if config and config.get('broker'):
  881. broker = config.get('broker')
  882. port = config.get('port', 1883)
  883. client_id = config.get('client_id', '')
  884. username = config.get('username', '')
  885. password = config.get('password', '')
  886. keepalive = config.get('keepalive', 60)
  887. topics = config.get('topics', [])
  888. logger.info(f"尝试自动连接MQTT: {broker}:{port}")
  889. will_topic = build_dtu_topic(dtu_config['customer_id'], 'dtu', dtu_config['dtu_id'], 'status')
  890. will_payload = json.dumps({
  891. 'dtu_id': dtu_config['dtu_id'], 'type': 'STATUS',
  892. 'payload': {'online': False, 'reason': 'CONNECTION_LOST'}
  893. })
  894. success, message = mqtt_client.connect(
  895. broker=broker, port=port, client_id=client_id,
  896. username=username or "", password=password or "",
  897. keepalive=keepalive,
  898. will_topic=will_topic, will_payload=will_payload, will_qos=1, will_retain=False
  899. )
  900. if success:
  901. logger.info(f"自动连接MQTT成功: {broker}:{port}")
  902. if topics:
  903. mqtt_client.subscribe(topics)
  904. else:
  905. logger.warning(f"自动连接MQTT失败: {message}")
  906. return success
  907. return False
  908. def auto_connect_serial():
  909. """自动连接上次使用的串口"""
  910. config = load_serial_config()
  911. if config and config.get('port'):
  912. port = config.get('port')
  913. baudrate = config.get('baudrate', 9600)
  914. timeout = config.get('timeout', 1.0)
  915. bytesize = config.get('bytesize', 8)
  916. parity = config.get('parity', 'N')
  917. stopbits = config.get('stopbits', 1)
  918. logger.info(f"尝试自动连接串口: {port} @ {baudrate}")
  919. success, message = serial_client.connect(port, baudrate=baudrate, timeout=timeout, bytesize=bytesize, parity=parity, stopbits=stopbits)
  920. if success:
  921. logger.info(f"自动连接串口成功: {port}")
  922. return True
  923. else:
  924. logger.warning(f"自动连接串口失败: {message}")
  925. return False
  926. # 客户端连接管理
  927. connected_clients = {
  928. 'data': set(),
  929. 'status': set(),
  930. 'control': set()
  931. }
  932. # 设置回调函数
  933. def serial_data_handler(data):
  934. """处理串口接收的数据"""
  935. try:
  936. timestamp = time.strftime('%Y-%m-%d %H:%M:%S')
  937. serial_data_buffer.append({
  938. 'timestamp': timestamp,
  939. 'data': data,
  940. 'direction': 'in'
  941. })
  942. if len(serial_data_buffer) > MAX_BUFFER_SIZE:
  943. serial_data_buffer.pop(0)
  944. socketio.emit('serial_data', {
  945. 'timestamp': timestamp,
  946. 'data': data,
  947. 'direction': 'in'
  948. }, namespace=SOCKETIO_NAMESPACE_DATA)
  949. if forward_serial_to_mqtt and _mqtt_connected():
  950. success, msg = mqtt_client.publish(mqtt_publish_topic, data)
  951. if not success:
  952. logger.warning(f"串口数据转发到MQTT失败: {msg}")
  953. except Exception as e:
  954. logger.error(f"处理串口数据时出错: {str(e)}")
  955. def serial_send_handler(data):
  956. """处理串口发送的数据"""
  957. try:
  958. timestamp = time.strftime('%Y-%m-%d %H:%M:%S')
  959. serial_data_buffer.append({
  960. 'timestamp': timestamp,
  961. 'data': data,
  962. 'direction': 'out'
  963. })
  964. if len(serial_data_buffer) > MAX_BUFFER_SIZE:
  965. serial_data_buffer.pop(0)
  966. socketio.emit('serial_data', {
  967. 'timestamp': timestamp,
  968. 'data': data,
  969. 'direction': 'out'
  970. }, namespace=SOCKETIO_NAMESPACE_DATA)
  971. except Exception as e:
  972. logger.error(f"处理串口发送数据时出错: {str(e)}")
  973. def serial_status_handler(status):
  974. """处理串口状态变化"""
  975. try:
  976. global serial_status
  977. serial_status = status
  978. # 通过WebSocket广播状态变化
  979. socketio.emit('serial_status', {
  980. 'connected': status
  981. }, namespace=SOCKETIO_NAMESPACE_STATUS)
  982. logger.info(f"串口状态更新: {'已连接' if status else '已断开'}")
  983. # 串口重新连接后,按线架绑定表重建面板配置(实际在线状态由地址段扫描刷新)
  984. if status:
  985. try:
  986. sync_panel_config()
  987. except Exception as e:
  988. logger.error(f"串口重连后同步面板配置失败: {e}")
  989. except Exception as e:
  990. logger.error(f"处理串口状态时出错: {str(e)}")
  991. def mqtt_data_handler(data):
  992. """处理MQTT接收的数据"""
  993. try:
  994. # 添加到缓冲区
  995. timestamp = time.strftime('%Y-%m-%d %H:%M:%S')
  996. mqtt_data_buffer.append({
  997. 'timestamp': timestamp,
  998. 'topic': data['topic'],
  999. 'payload': data['payload']
  1000. })
  1001. # 保持缓冲区大小
  1002. if len(mqtt_data_buffer) > MAX_BUFFER_SIZE:
  1003. mqtt_data_buffer.pop(0)
  1004. # 通过WebSocket广播数据
  1005. socketio.emit('mqtt_data', {
  1006. 'timestamp': timestamp,
  1007. 'topic': data['topic'],
  1008. 'payload': data['payload']
  1009. }, namespace=SOCKETIO_NAMESPACE_DATA)
  1010. # 如果启用了转发且串口已连接,转发数据到串口
  1011. if forward_mqtt_to_serial and _serial_connected():
  1012. success, msg = serial_client.send_data(data['payload'])
  1013. if not success:
  1014. logger.warning(f"MQTT数据转发到串口失败: {msg}")
  1015. except Exception as e:
  1016. logger.error(f"处理MQTT数据时出错: {str(e)}")
  1017. def mqtt_status_handler(status):
  1018. """处理MQTT状态变化"""
  1019. try:
  1020. global mqtt_status
  1021. mqtt_status = status
  1022. # 通过WebSocket广播状态变化
  1023. socketio.emit('mqtt_status', {
  1024. 'connected': status
  1025. }, namespace=SOCKETIO_NAMESPACE_STATUS)
  1026. logger.info(f"MQTT状态更新: {'已连接' if status else '已断开'}")
  1027. # 如果MQTT连接成功且启用了DTU协议,发送注册消息和订阅控制主题
  1028. if status and dtu_config.get('enabled'):
  1029. socketio.sleep(1) # 等待连接稳定
  1030. # 发送DTU注册消息
  1031. dtu_register()
  1032. # 订阅控制主题
  1033. control_topic = build_dtu_topic(dtu_config['customer_id'], 'dtu', dtu_config['dtu_id'], 'control')
  1034. mqtt_client.subscribe(control_topic, qos=1)
  1035. logger.info(f"已订阅控制主题: {control_topic}")
  1036. # 订阅广播主题 (DTU发现 7.14 和批量配置 7.15)
  1037. discover_topic = build_dtu_topic('broadcast', 'dtu', 'discover')
  1038. config_topic = build_dtu_topic('broadcast', 'dtu', 'config')
  1039. mqtt_client.subscribe(discover_topic, qos=1)
  1040. mqtt_client.subscribe(config_topic, qos=1)
  1041. logger.info(f"已订阅广播主题: {discover_topic}, {config_topic}")
  1042. # MQTT 状态变化时刷新当前屏幕显示
  1043. panels = get_sorted_panels()
  1044. if screen_display.get_status().get('connected'):
  1045. try:
  1046. if panels:
  1047. with screen_lock:
  1048. current_idx = screen_current_panel_index
  1049. if 0 <= current_idx < len(panels):
  1050. screen_refresh_queue.put((panels[current_idx][0],))
  1051. else:
  1052. screen_refresh_queue.put((None,))
  1053. except Exception as e:
  1054. logger.error(f"MQTT 状态变化刷新屏幕失败: {e}")
  1055. except Exception as e:
  1056. logger.error(f"处理MQTT状态时出错: {str(e)}")
  1057. # ========== DTU MQTT协议处理函数 ==========
  1058. def build_dtu_topic(*parts):
  1059. """构建DTU MQTT主题"""
  1060. prefix = dtu_config.get('topic_prefix', '线架系统')
  1061. return '/'.join([prefix] + list(parts))
  1062. def dtu_register(discovery_request_id=None):
  1063. """发送DTU注册消息"""
  1064. if not _mqtt_connected():
  1065. logger.warning("MQTT未连接,无法发送注册消息")
  1066. return False
  1067. try:
  1068. # 加载面板配置
  1069. devices = address_config.get_stored_devices()
  1070. # 优先使用 dtu_config 中预设的 panel_id 映射 (uid_hex -> panel_id)
  1071. panel_id_map = dtu_config.get('panel_id_mapping', {}) or {}
  1072. panels = []
  1073. panel_id = 1
  1074. for uid_hex, addr in devices.items():
  1075. custom_id = panel_id_map.get(uid_hex) or panel_id_map.get(addr)
  1076. if custom_id:
  1077. pid = str(custom_id)
  1078. else:
  1079. pid = f"PANEL_{dtu_config['dtu_id']}_{addr}"
  1080. panels.append({
  1081. 'panel_id': pid,
  1082. 'address': addr,
  1083. 'position': panel_id,
  1084. 'panel_uid': uid_hex
  1085. })
  1086. panel_id += 1
  1087. # 检测网络类型
  1088. network_type = 'ethernet'
  1089. signal_strength = None
  1090. try:
  1091. interfaces = network_manager.get_network_status().get('interfaces', {})
  1092. for name, info in interfaces.items():
  1093. if name in ('lo', 'docker0') or name.startswith(('br-', 'veth')):
  1094. continue
  1095. if info.get('status') != 'UP':
  1096. continue
  1097. if name.startswith(('wlan', 'wlp')):
  1098. network_type = 'wifi'
  1099. # 读取信号强度 (dBm, 越接近 0 越强)
  1100. try:
  1101. import re
  1102. with open('/proc/net/wireless') as f:
  1103. for line in f.readlines()[2:]:
  1104. m = re.match(r'\s*(\S+):.*\s(\d+)\.\s+(-?\d+)\.', line)
  1105. if m:
  1106. signal_strength = int(m.group(3))
  1107. break
  1108. except Exception:
  1109. pass
  1110. elif name.startswith(('wwan', 'ppp', 'usb', 'eth')):
  1111. # 移动网卡
  1112. if name.startswith(('wwan', 'ppp', 'usb')):
  1113. network_type = 'cellular'
  1114. break
  1115. except Exception:
  1116. pass
  1117. payload = {
  1118. 'msg_id': f"reg_{int(time.time() * 1000)}_{uuid.uuid4().hex[:6]}",
  1119. 'timestamp': int(time.time() * 1000),
  1120. 'dtu_id': dtu_config['dtu_id'],
  1121. 'type': 'REGISTER',
  1122. 'payload': {
  1123. 'firmware_version': dtu_config.get('firmware_version', 'v1.0.0'),
  1124. 'hardware_version': dtu_config.get('hardware_version', 'v1.0'),
  1125. 'panel_count': len(panels),
  1126. 'network_type': network_type,
  1127. 'signal_strength': signal_strength,
  1128. 'uptime': int((time.time() - _boot_time) * 1000),
  1129. 'panels': panels,
  1130. 'discovery_request_id': discovery_request_id
  1131. }
  1132. }
  1133. if discovery_request_id:
  1134. payload['payload']['discovery_request_id'] = discovery_request_id
  1135. topic = build_dtu_topic(dtu_config['customer_id'], 'dtu', dtu_config['dtu_id'], 'register')
  1136. success, msg = mqtt_client.publish(topic, json.dumps(payload), qos=1)
  1137. if success:
  1138. logger.info(f"DTU注册消息已发送: {topic}")
  1139. else:
  1140. logger.error(f"DTU注册消息发送失败: {msg}")
  1141. return success
  1142. except Exception as e:
  1143. logger.error(f"发送DTU注册消息失败: {str(e)}")
  1144. return False
  1145. _discover_dedup = {}
  1146. def handle_broadcast_discover(payload):
  1147. """处理广播发现消息 (7.14)"""
  1148. try:
  1149. if not isinstance(payload, dict) or payload.get('type') != 'DISCOVER':
  1150. return
  1151. p = payload.get('payload', {})
  1152. cust_id = p.get('customer_id', '')
  1153. if cust_id and cust_id != dtu_config.get('customer_id'):
  1154. return
  1155. request_id = p.get('request_id', '')
  1156. now = time.time()
  1157. if request_id and request_id in _discover_dedup:
  1158. if now - _discover_dedup[request_id] < 60:
  1159. return
  1160. _discover_dedup[request_id] = now
  1161. # 清理超过 60s 的旧条目,防止内存泄漏
  1162. if len(_discover_dedup) > 256:
  1163. for k in list(_discover_dedup.keys()):
  1164. if now - _discover_dedup[k] > 60:
  1165. _discover_dedup.pop(k, None)
  1166. logger.info(f"收到DISCOVER广播,响应注册消息 (request_id={request_id})")
  1167. import random
  1168. delay = random.uniform(0, 3.0)
  1169. time.sleep(delay)
  1170. dtu_register(discovery_request_id=request_id)
  1171. dtu_publish_status()
  1172. except Exception as e:
  1173. logger.error(f"处理广播发现消息失败: {str(e)}")
  1174. def handle_broadcast_config(payload):
  1175. """处理广播批量配置消息 (7.15)"""
  1176. try:
  1177. if isinstance(payload, dict) and payload.get('type') == 'CONFIG':
  1178. cfg = payload.get('payload', {})
  1179. cust_id = cfg.get('customer_id', '')
  1180. if cust_id and cust_id != dtu_config.get('customer_id'):
  1181. return
  1182. config_version = cfg.get('config_version', '')
  1183. force_apply = cfg.get('force_apply', False)
  1184. items = cfg.get('items', {})
  1185. local_version = dtu_config.get('config_version', '')
  1186. incoming_msg_id = payload.get('msg_id', '')
  1187. if local_version == config_version and not force_apply:
  1188. logger.info(f"配置版本 {config_version} 已存在,跳过")
  1189. _send_config_response(config_version, [], True, 1011, incoming_msg_id)
  1190. return
  1191. valid_ranges = {
  1192. 'heartbeat_interval': (10, 3600),
  1193. 'poll_interval_ms': (100, 10000),
  1194. 'modbus_timeout_ms': (200, 5000),
  1195. 'mqtt_keepalive': (10, 300),
  1196. 'event_debounce_ms': (0, 5000),
  1197. }
  1198. applied = []
  1199. for key, val in items.items():
  1200. if key in valid_ranges:
  1201. lo, hi = valid_ranges[key]
  1202. if not (lo <= val <= hi):
  1203. _send_config_response(config_version, applied, False, 1012, incoming_msg_id)
  1204. return
  1205. if key == 'heartbeat_interval':
  1206. dtu_config['heartbeat_interval'] = val
  1207. elif key == 'mqtt_keepalive':
  1208. dtu_config['mqtt_keepalive'] = val
  1209. elif key == 'event_debounce_ms':
  1210. dtu_config['event_debounce_ms'] = val
  1211. elif key == 'poll_interval_ms':
  1212. dtu_config['poll_interval_ms'] = val
  1213. elif key == 'modbus_timeout_ms':
  1214. dtu_config['modbus_timeout_ms'] = val
  1215. applied.append(key)
  1216. elif key in ('alarm_enabled',):
  1217. dtu_config['alarm_enabled'] = bool(val)
  1218. applied.append(key)
  1219. else:
  1220. _send_config_response(config_version, applied, False, 1013, incoming_msg_id)
  1221. return
  1222. if applied:
  1223. dtu_config['config_version'] = config_version
  1224. try:
  1225. save_dtu_config()
  1226. except Exception as e:
  1227. logger.error(f"配置写入持久化失败: {e}")
  1228. _send_config_response(config_version, applied, False, 1014, incoming_msg_id)
  1229. return
  1230. _send_config_response(config_version, applied, True, 0, incoming_msg_id)
  1231. except Exception as e:
  1232. logger.error(f"处理广播配置消息失败: {str(e)}")
  1233. def _send_config_response(config_version, applied_items, success, error_code, original_msg_id=''):
  1234. """发送批量配置响应 (随机 0-2000ms 延迟避免广播风暴)"""
  1235. import time as _t, json as _j, random as _r
  1236. delay = _r.uniform(0, 2.0)
  1237. def _delayed_publish():
  1238. _t.sleep(delay)
  1239. topic = build_dtu_topic(dtu_config['customer_id'], 'dtu', dtu_config['dtu_id'], 'response')
  1240. payload = {
  1241. 'msg_id': f"rsp_cfg_{int(_t.time() * 1000)}",
  1242. 'timestamp': int(_t.time() * 1000),
  1243. 'dtu_id': dtu_config['dtu_id'],
  1244. 'type': 'RESPONSE',
  1245. 'payload': {
  1246. 'original_msg_id': original_msg_id or f"cfg_{config_version}",
  1247. 'command': 'BROADCAST_CONFIG',
  1248. 'success': success,
  1249. 'result': {
  1250. 'applied_version': config_version,
  1251. 'applied_items': applied_items,
  1252. 'restart_required': False
  1253. } if success else None,
  1254. 'error_code': error_code,
  1255. 'error_message': get_ota_error_message(error_code) if error_code and error_code >= 1000 else None
  1256. }
  1257. }
  1258. mqtt_client.publish(topic, _j.dumps(payload), qos=1)
  1259. # 异步执行避免阻塞 MQTT 主循环
  1260. threading.Thread(target=_delayed_publish, daemon=True).start()
  1261. def dtu_publish_status(force=False):
  1262. """发送DTU状态/心跳消息
  1263. Args:
  1264. force: 保留参数。事件/告警/查询触发时传入 True,用于立即上报(区别于定时心跳)
  1265. """
  1266. if not _mqtt_connected() or not dtu_config.get('enabled'):
  1267. return False
  1268. # 调试日志: 区分定时 vs 触发上报
  1269. if force:
  1270. logger.debug("dtu_publish_status(force=True) - 事件触发立即上报")
  1271. try:
  1272. # panel_online 计数: 在最近 60s 内所有 24 个端口都被成功轮询过
  1273. # 严格语义: 全部 24 端口在最近一轮 poll 周期内被 poll 成功 → online
  1274. now_ms = int(time.time() * 1000)
  1275. threshold_ms = 60_000 # 60 秒未轮询视为 offline
  1276. poll_ok = 0
  1277. for pid, cfg in panel_config.items():
  1278. ps = port_state.get(pid, {})
  1279. expected_ports = cfg.get('port_count', 24)
  1280. polled_recently = sum(
  1281. 1 for p in ps.values()
  1282. if p.get('last_polled_at') and (now_ms - p.get('last_polled_at', 0)) < threshold_ms
  1283. )
  1284. if polled_recently >= expected_ports:
  1285. poll_ok += 1
  1286. panel_online = poll_ok
  1287. panel_offline = max(0, len(panel_config) - panel_online)
  1288. # 检查串口状态
  1289. serial_st = serial_client.get_status()
  1290. rs485_status = 'normal' if (isinstance(serial_st, dict) and serial_st.get('connected', False)) else 'error'
  1291. # 网络状态: 检查实际接口 UP 状态
  1292. # 规则: 有线 (eth*/wlan*) UP → online; 仅 wifi 信号弱 (无 link/低质量) → weak; 否则 offline
  1293. network_status = 'offline'
  1294. try:
  1295. interfaces = network_manager.get_network_status().get('interfaces', {})
  1296. real_ifs = {i: s for i, s in interfaces.items()
  1297. if s.get('status') == 'UP'
  1298. and i not in ('lo', 'docker0')
  1299. and not i.startswith(('br-', 'veth'))}
  1300. if real_ifs:
  1301. # 优先有线连接
  1302. if any(i.startswith(('eth', 'en')) for i in real_ifs):
  1303. network_status = 'online'
  1304. else:
  1305. # 仅无线: 检查信号强度 (Linux: /proc/net/wireless)
  1306. try:
  1307. import re
  1308. with open('/proc/net/wireless') as f:
  1309. lines = f.readlines()
  1310. quality = None
  1311. for line in lines[2:]: # 跳过表头
  1312. m = re.match(r'\s*(\S+):.*\s(\d+)\.\s+(\d+)\.\s+(\d+)', line)
  1313. if m:
  1314. # link quality (字段 2, 0-70); signal level (字段 3, dBm, 越低越差)
  1315. link = int(m.group(2))
  1316. sig = int(m.group(3))
  1317. # 估算: link < 30 或 sig < -75 dBm 视为弱
  1318. if quality is None or link < quality:
  1319. quality = link
  1320. network_status = 'weak' if (quality is not None and quality < 30) else 'online'
  1321. except Exception:
  1322. network_status = 'online'
  1323. except Exception:
  1324. network_status = 'offline'
  1325. # 屏幕连接状态:从 Nextion 屏幕实例读取
  1326. screen_connected = screen_display.get_status().get('connected', False)
  1327. # CPU/内存使用率 (Linux 才有意义)
  1328. cpu_usage = _get_cpu_usage()
  1329. memory_usage = _get_memory_usage()
  1330. dtu_temperature = None
  1331. try:
  1332. # 主板温度: 优先 psutil.sensors_temperatures
  1333. try:
  1334. import psutil
  1335. temps = psutil.sensors_temperatures(fahrenheit=False) or {}
  1336. for chip_name, entries in temps.items():
  1337. for entry in entries:
  1338. if entry.current and entry.current > 0:
  1339. dtu_temperature = round(entry.current, 1)
  1340. break
  1341. if dtu_temperature is not None:
  1342. break
  1343. except Exception:
  1344. pass
  1345. # 回退: 直接读 /sys/class/thermal/thermal_zone*/temp
  1346. if dtu_temperature is None:
  1347. import glob as _glob
  1348. for tz in sorted(_glob.glob('/sys/class/thermal/thermal_zone*/temp')):
  1349. try:
  1350. with open(tz) as f:
  1351. v = int(f.read().strip())
  1352. if v > 0:
  1353. dtu_temperature = round(v / 1000.0, 1)
  1354. break
  1355. except Exception:
  1356. continue
  1357. except Exception:
  1358. pass
  1359. # 优先使用外部上报的 dtu_temperature (来自 MQTT/RS485 传感器)
  1360. dtu_temperature = env_sensor_data.get('dtu_temperature') or dtu_temperature
  1361. payload = {
  1362. 'msg_id': f"hbt_{int(time.time() * 1000)}_{uuid.uuid4().hex[:6]}",
  1363. 'timestamp': int(time.time() * 1000),
  1364. 'dtu_id': dtu_config['dtu_id'],
  1365. 'type': 'STATUS',
  1366. 'payload': {
  1367. 'cpu_usage': cpu_usage,
  1368. 'memory_usage': memory_usage,
  1369. 'temperature': dtu_temperature,
  1370. 'network_status': network_status,
  1371. 'panel_online': panel_online,
  1372. 'panel_offline': panel_offline,
  1373. 'screen_connected': screen_connected,
  1374. 'mqtt_connected': _mqtt_connected(),
  1375. 'rs485_status': rs485_status
  1376. }
  1377. }
  1378. topic = build_dtu_topic(dtu_config['customer_id'], 'dtu', dtu_config['dtu_id'], 'status')
  1379. success, _ = mqtt_client.publish(topic, json.dumps(payload), qos=1)
  1380. return success
  1381. except Exception as e:
  1382. logger.error(f"发送DTU状态消息失败: {str(e)}")
  1383. return False
  1384. def dtu_publish_event(panel_id, port_id, event_type, jumper_uid, previous_jumper_uid=None):
  1385. """发送端口事件消息。
  1386. 入参 jumper_uid / previous_jumper_uid 为内部 12 位硬件卡号;
  1387. MQTT 上行拼成 16 位(前 4 位 DTU 号 + 后 12 位卡号)。
  1388. """
  1389. if not _mqtt_connected() or not dtu_config.get('enabled'):
  1390. return False
  1391. try:
  1392. mqtt_jumper = _to_mqtt_jumper_uid(jumper_uid)
  1393. mqtt_prev = _to_mqtt_jumper_uid(previous_jumper_uid)
  1394. payload = {
  1395. 'msg_id': f"evt_{int(time.time() * 1000)}",
  1396. 'timestamp': int(time.time() * 1000),
  1397. 'dtu_id': dtu_config['dtu_id'],
  1398. 'type': 'EVENT',
  1399. 'payload': {
  1400. 'panel_id': panel_id,
  1401. 'port_id': port_id,
  1402. 'event_type': event_type,
  1403. 'event_id': f"evt_{uuid.uuid4().hex[:8]}",
  1404. 'jumper_uid': mqtt_jumper,
  1405. 'previous_jumper_uid': mqtt_prev
  1406. }
  1407. }
  1408. topic = build_dtu_topic(dtu_config['customer_id'], 'patchpanel', dtu_config['dtu_id'], panel_id, 'event')
  1409. success, _ = mqtt_client.publish(topic, json.dumps(payload), qos=1)
  1410. if success:
  1411. logger.info(f"端口事件已发送: {panel_id}:{port_id} - {event_type}")
  1412. # 记录到事件历史(保留内部 12 位,供本地前端/调试)
  1413. port_event_history.append({
  1414. 'timestamp': int(time.time() * 1000),
  1415. 'panel_id': panel_id,
  1416. 'port_id': port_id,
  1417. 'event_type': event_type,
  1418. 'jumper_uid': jumper_uid,
  1419. 'previous_jumper_uid': previous_jumper_uid
  1420. })
  1421. # 限制历史记录数量
  1422. if len(port_event_history) > 1000:
  1423. port_event_history[:] = port_event_history[-1000:]
  1424. # 事件驱动: 立即上报一次 status (符合 7.4 "状态变化时立即上报")
  1425. try:
  1426. dtu_publish_status(force=True)
  1427. except Exception:
  1428. pass
  1429. return success
  1430. except Exception as e:
  1431. logger.error(f"发送端口事件失败: {str(e)}")
  1432. return False
  1433. def dtu_publish_alarm(panel_id, port_id, alarm_type, expected_jumper_uid, actual_jumper_uid, severity='WARNING'):
  1434. """发送非法告警消息。
  1435. 入参 expected_jumper_uid / actual_jumper_uid 为内部 12 位硬件卡号;
  1436. MQTT 上行拼成 16 位(前 4 位 DTU 号 + 后 12 位卡号)。
  1437. """
  1438. if not _mqtt_connected() or not dtu_config.get('enabled'):
  1439. return False
  1440. try:
  1441. mqtt_exp = _to_mqtt_jumper_uid(expected_jumper_uid)
  1442. mqtt_act = _to_mqtt_jumper_uid(actual_jumper_uid)
  1443. description = f"端口{port_id}期望跳线{mqtt_exp or '无'},实际{'未读到' if not actual_jumper_uid else mqtt_act}"
  1444. payload = {
  1445. 'msg_id': f"alm_{int(time.time() * 1000)}",
  1446. 'timestamp': int(time.time() * 1000),
  1447. 'dtu_id': dtu_config['dtu_id'],
  1448. 'type': 'ALARM',
  1449. 'payload': {
  1450. 'panel_id': panel_id,
  1451. 'port_id': port_id,
  1452. 'alarm_type': alarm_type,
  1453. 'severity': severity,
  1454. 'expected_jumper_uid': mqtt_exp,
  1455. 'actual_jumper_uid': mqtt_act,
  1456. 'description': description
  1457. }
  1458. }
  1459. topic = build_dtu_topic(dtu_config['customer_id'], 'patchpanel', dtu_config['dtu_id'], panel_id, 'alarm')
  1460. success, _ = mqtt_client.publish(topic, json.dumps(payload), qos=1)
  1461. if success:
  1462. logger.warning(f"非法告警已发送: {panel_id}:{port_id} - {alarm_type}")
  1463. # 记录到事件历史(保留内部 12 位,供本地前端/调试)
  1464. port_event_history.append({
  1465. 'timestamp': int(time.time() * 1000),
  1466. 'panel_id': panel_id,
  1467. 'port_id': port_id,
  1468. 'event_type': 'ALARM',
  1469. 'alarm_type': alarm_type,
  1470. 'expected_jumper_uid': expected_jumper_uid,
  1471. 'actual_jumper_uid': actual_jumper_uid
  1472. })
  1473. # 限制历史记录数量
  1474. if len(port_event_history) > 1000:
  1475. port_event_history[:] = port_event_history[-1000:]
  1476. # 事件驱动: 告警后立即上报一次 status
  1477. try:
  1478. dtu_publish_status(force=True)
  1479. except Exception:
  1480. pass
  1481. return success
  1482. except Exception as e:
  1483. logger.error(f"发送非法告警失败: {str(e)}")
  1484. return False
  1485. def dtu_publish_panel_status(panel_id, address, ports_data, online=True):
  1486. """发送配线架状态消息 (7.7)
  1487. online=False 时表示面板已离线,ports 全部置为 UNKNOWN。
  1488. """
  1489. if not _mqtt_connected() or not dtu_config.get('enabled'):
  1490. return False
  1491. try:
  1492. if not online:
  1493. ports_data = [{'port_id': p, 'status': 'UNKNOWN', 'jumper_uid': None} for p in range(1, 25)]
  1494. # MQTT 上行 jumper_uid 拼成 16 位(前 4 位 DTU 号 + 后 12 位卡号)
  1495. mqtt_ports = [
  1496. {**p, 'jumper_uid': _to_mqtt_jumper_uid(p.get('jumper_uid'))}
  1497. for p in ports_data
  1498. ]
  1499. topic = build_dtu_topic(dtu_config['customer_id'], 'patchpanel', dtu_config['dtu_id'], panel_id, 'status')
  1500. payload = {
  1501. 'msg_id': f"pst_{int(time.time() * 1000)}",
  1502. 'timestamp': int(time.time() * 1000),
  1503. 'dtu_id': dtu_config['dtu_id'],
  1504. 'type': 'STATUS',
  1505. 'payload': {
  1506. 'panel_id': panel_id,
  1507. 'position': panel_config.get(panel_id, {}).get('position', 0),
  1508. 'address': address,
  1509. 'panel_uid': panel_config.get(panel_id, {}).get('panel_uid'),
  1510. 'online': online,
  1511. 'last_poll_time': int(time.time() * 1000),
  1512. 'ports': mqtt_ports
  1513. }
  1514. }
  1515. mqtt_client.publish(topic, json.dumps(payload), qos=0)
  1516. return True
  1517. except Exception as e:
  1518. logger.error(f"发送面板状态失败: {str(e)}")
  1519. return False
  1520. def dtu_publish_jumper_status():
  1521. """发送跳线状态汇总 (7.8) - STATUS envelope with panels[] array"""
  1522. if not _mqtt_connected() or not dtu_config.get('enabled'):
  1523. return False
  1524. try:
  1525. panels_arr = []
  1526. for panel_id, cfg in panel_config.items():
  1527. ports_arr = []
  1528. ps = port_state.get(panel_id, {})
  1529. for port_id in range(1, 25):
  1530. p = ps.get(port_id, {})
  1531. last_uid = p.get('last_uid')
  1532. expected_uid = p.get('expected_uid')
  1533. last_polled = p.get('last_polled_at')
  1534. if not last_polled:
  1535. status = 'UNKNOWN'
  1536. elif last_uid is None:
  1537. status = 'DISCONNECTED'
  1538. if expected_uid:
  1539. status = 'ILLEGAL'
  1540. elif expected_uid and last_uid != expected_uid:
  1541. status = 'ILLEGAL'
  1542. else:
  1543. status = 'CONNECTED'
  1544. ports_arr.append({
  1545. 'port_id': port_id,
  1546. 'status': status,
  1547. # MQTT 上行拼成 16 位(前 4 位 DTU 号 + 后 12 位卡号)
  1548. 'jumper_uid': _to_mqtt_jumper_uid(last_uid)
  1549. })
  1550. panels_arr.append({
  1551. 'panel_id': panel_id,
  1552. 'position': cfg.get('position', 0),
  1553. 'ports': ports_arr
  1554. })
  1555. topic = build_dtu_topic(dtu_config['customer_id'], 'jumper', dtu_config['dtu_id'], 'status')
  1556. payload = {
  1557. 'msg_id': f"jmp_{int(time.time() * 1000)}",
  1558. 'timestamp': int(time.time() * 1000),
  1559. 'dtu_id': dtu_config['dtu_id'],
  1560. 'type': 'STATUS',
  1561. 'payload': {'panels': panels_arr}
  1562. }
  1563. mqtt_client.publish(topic, json.dumps(payload), qos=0)
  1564. return True
  1565. except Exception as e:
  1566. logger.error(f"发送跳线状态汇总失败: {str(e)}")
  1567. return False
  1568. _env_sensor_data = {
  1569. 'temperature': None,
  1570. 'humidity': None,
  1571. 'sensor_update_time': None
  1572. }
  1573. def update_env_sensor_data(temperature, humidity, dtu_temperature=None, sensor_update_time=None):
  1574. """更新环境传感器数据 - 委托给完整实现(带历史和告警)"""
  1575. update_env_sensor_data_full(temperature, humidity, dtu_temperature, sensor_update_time)
  1576. def dtu_publish_env_sensor():
  1577. """发送环境传感器数据 (7.9) - STATUS envelope"""
  1578. if not _mqtt_connected() or not dtu_config.get('enabled'):
  1579. return False
  1580. try:
  1581. topic = build_dtu_topic(dtu_config['customer_id'], 'env', dtu_config['dtu_id'], 'sensor')
  1582. payload = {
  1583. 'msg_id': f"sen_{int(time.time() * 1000)}",
  1584. 'timestamp': int(time.time() * 1000),
  1585. 'dtu_id': dtu_config['dtu_id'],
  1586. 'type': 'STATUS',
  1587. 'payload': {
  1588. 'temperature': _env_sensor_data.get('temperature'),
  1589. 'humidity': _env_sensor_data.get('humidity'),
  1590. 'sensor_update_time': _env_sensor_data.get('sensor_update_time')
  1591. }
  1592. }
  1593. mqtt_client.publish(topic, json.dumps(payload), qos=0)
  1594. return True
  1595. except Exception as e:
  1596. logger.error(f"发送环境传感器数据失败: {str(e)}")
  1597. return False
  1598. def _resolve_panel_id(target):
  1599. """将 target 解析为 panel_config 中实际的 panel_id 键。
  1600. 兼容三种输入:
  1601. - 完整 panel_id(如 'PANEL_dtu_001_2'):原样返回(若存在)
  1602. - 数字地址/序号(如 2 或 '2'):尝试匹配 panel_config 中以该数字结尾的键
  1603. - 'all' / None:原样返回(由调用方处理)
  1604. """
  1605. if target is None or target == 'all':
  1606. return target
  1607. if target in panel_config:
  1608. return target
  1609. # 数字型:尝试匹配 PANEL_*_{n} 或地址为 n 的面板
  1610. try:
  1611. num = int(target)
  1612. except (TypeError, ValueError):
  1613. return target
  1614. # 1) 键以 _{num} 结尾
  1615. for pid in panel_config:
  1616. if pid.endswith(f'_{num}'):
  1617. return pid
  1618. # 2) 面板地址等于 num
  1619. for pid, cfg in panel_config.items():
  1620. if cfg.get('address') == num:
  1621. return pid
  1622. return target
  1623. def dtu_handle_control(topic, payload):
  1624. """处理下行控制指令"""
  1625. try:
  1626. if not dtu_config.get('enabled'):
  1627. return
  1628. command = payload.get('payload', {}).get('command')
  1629. target = payload.get('payload', {}).get('target')
  1630. params = payload.get('payload', {}).get('params', {})
  1631. logger.info(f"收到控制指令: command={command}, target={target}")
  1632. response_payload = {
  1633. 'msg_id': f"rsp_{int(time.time() * 1000)}_{uuid.uuid4().hex[:6]}",
  1634. 'timestamp': int(time.time() * 1000),
  1635. 'dtu_id': dtu_config['dtu_id'],
  1636. 'type': 'RESPONSE',
  1637. 'payload': {
  1638. 'original_msg_id': payload.get('msg_id'),
  1639. 'command': command,
  1640. 'success': True,
  1641. 'result': {},
  1642. 'error_code': 0,
  1643. 'error_message': None
  1644. }
  1645. }
  1646. topic_response_dtu = build_dtu_topic(
  1647. dtu_config['customer_id'], 'dtu', dtu_config['dtu_id'], 'response')
  1648. # 处理各命令
  1649. if command == 'SET_PORT_LED':
  1650. # 设置端口LED
  1651. port_id = params.get('port_id')
  1652. led_mode = params.get('led_mode', 'OFF')
  1653. # 参数校验
  1654. if port_id is None or not (1 <= int(port_id) <= 24):
  1655. response_payload['payload']['success'] = False
  1656. response_payload['payload']['error_code'] = 1003
  1657. response_payload['payload']['error_message'] = f"端口号非法: {port_id}"
  1658. mqtt_client.publish(topic_response_dtu, json.dumps(response_payload), qos=1)
  1659. return
  1660. if led_mode not in ('OFF', 'BLINK_RED', 'BLINK_GREEN', 'BLINK_BLUE'):
  1661. response_payload['payload']['success'] = False
  1662. response_payload['payload']['error_code'] = 1004
  1663. response_payload['payload']['error_message'] = f"LED模式非法: {led_mode}"
  1664. mqtt_client.publish(topic_response_dtu, json.dumps(response_payload), qos=1)
  1665. return
  1666. # LED模式映射
  1667. led_mode_map = {'OFF': 0, 'BLINK_RED': 1, 'BLINK_GREEN': 2, 'BLINK_BLUE': 3}
  1668. color = led_mode_map.get(led_mode, 0)
  1669. # 查找设备地址
  1670. device_address = 1 # 默认
  1671. for panel_id, cfg in panel_config.items():
  1672. if panel_id == target:
  1673. device_address = cfg.get('address', 1)
  1674. break
  1675. result = modbus_client.set_rgb_led(device_address, port_id, color)
  1676. response_payload['payload']['success'] = 'error' not in result
  1677. response_payload['payload']['result'] = result
  1678. if 'error' in result:
  1679. response_payload['payload']['error_code'] = 1005
  1680. response_payload['payload']['error_message'] = result['error']
  1681. else:
  1682. if device_address not in led_states:
  1683. led_states[device_address] = {}
  1684. led_states[device_address][int(port_id)] = color
  1685. _refresh_current_screen_panel()
  1686. elif command == 'QUERY_DTU_STATUS':
  1687. # 查询DTU状态
  1688. dtu_publish_status(force=True)
  1689. elif command == 'SYNC_PORT_MAPPING':
  1690. # 同步单端口期望映射
  1691. port_id = params.get('port_id')
  1692. jumper_uid = params.get('jumper_uid')
  1693. # 前端把 panel_id 放在 params 里(target='port' 占位),优先取 params.panel_id
  1694. panel_target = params.get('panel_id')
  1695. if panel_target is None:
  1696. panel_target = target
  1697. # 兼容数字 panel_id:前端可能传 2/3 或 'PANEL_dtu_001_2'
  1698. panel_target = _resolve_panel_id(panel_target)
  1699. if port_id is None or panel_target is None or panel_target not in panel_config:
  1700. response_payload['payload']['success'] = False
  1701. response_payload['payload']['error_code'] = 1006 if port_id is None else 1002
  1702. response_payload['payload']['error_message'] = "参数缺失" if port_id is None else f"目标面板不存在: {panel_target}"
  1703. mqtt_client.publish(topic_response_dtu, json.dumps(response_payload), qos=1)
  1704. return
  1705. try:
  1706. # MQTT 16 位 -> 内部 12 位硬件卡号(剥前 4 位 DTU 号)
  1707. expected_uid = _from_mqtt_jumper_uid(jumper_uid)
  1708. except ValueError as e:
  1709. response_payload['payload']['success'] = False
  1710. response_payload['payload']['error_code'] = 1006
  1711. response_payload['payload']['error_message'] = str(e)
  1712. mqtt_client.publish(topic_response_dtu, json.dumps(response_payload), qos=1)
  1713. return
  1714. if panel_target not in port_state:
  1715. port_state[panel_target] = {}
  1716. if port_id not in port_state[panel_target]:
  1717. port_state[panel_target][port_id] = {'last_uid': None, 'expected_uid': None, 'alarm_count': 0}
  1718. port_state[panel_target][port_id]['expected_uid'] = expected_uid
  1719. # 持久化期望映射(内部 12 位),保证离线恢复/重启后仍能判定非法连接
  1720. dtu_config.setdefault('port_mappings', {}).setdefault(panel_target, {})[str(port_id)] = expected_uid
  1721. save_dtu_config()
  1722. # V4: 下发卡号到配线架硬件(24路同时发送;保留其他端口真实硬件值,避免误擦)
  1723. device_address = panel_config[panel_target].get('address', 1)
  1724. # 策略同 HTTP 单端口模式:硬件读基底 → dtu_config 覆盖 → 目标端口覆盖
  1725. card_mappings = {i: None for i in range(1, 25)}
  1726. try:
  1727. hw_read = modbus_client.read_antenna_card(device_address, timeout=2.0)
  1728. if 'error' not in hw_read:
  1729. for card in hw_read.get('cards', []):
  1730. ant = card.get('antenna')
  1731. cs = card.get('card_str') or card.get('uid')
  1732. if ant and cs and str(cs).lower() != 'ffffffffffff':
  1733. cs_clean = str(cs).strip().upper().replace(':', '').replace(' ', '')
  1734. if len(cs_clean) == 12 and all(c in _JUMPER_HEX for c in cs_clean):
  1735. card_mappings[int(ant)] = cs_clean
  1736. except Exception as e:
  1737. logger.warning(f"SYNC_PORT_MAPPING 读取硬件基底失败: {e}")
  1738. # dtu_config 期望值覆盖(平台下发过的映射是权威)
  1739. dtu_base = _build_card_mappings_for_panel(panel_target)
  1740. for ant, v in dtu_base.items():
  1741. if v is not None:
  1742. card_mappings[ant] = v
  1743. # 目标端口覆盖(本次要设置/清除的值)
  1744. try:
  1745. card_mappings[int(port_id)] = expected_uid
  1746. except (ValueError, TypeError):
  1747. response_payload['payload']['success'] = False
  1748. response_payload['payload']['error_code'] = 1006
  1749. response_payload['payload']['error_message'] = f"port_id 非法: {port_id}"
  1750. mqtt_client.publish(topic_response_dtu, json.dumps(response_payload), qos=1)
  1751. return
  1752. try:
  1753. hw_result = modbus_client.set_card_mapping(device_address, card_mappings, timeout=2.0)
  1754. if 'error' in hw_result:
  1755. response_payload['payload']['success'] = False
  1756. response_payload['payload']['error_code'] = 1008
  1757. response_payload['payload']['error_message'] = f"卡号下发硬件失败: {hw_result['error']}"
  1758. mqtt_client.publish(topic_response_dtu, json.dumps(response_payload), qos=1)
  1759. return
  1760. response_payload['payload']['result'] = {'hardware': hw_result}
  1761. logger.info(f"SYNC_PORT_MAPPING 面板{panel_target}端口{port_id} 卡号已下发硬件: {expected_uid}")
  1762. except Exception as e:
  1763. response_payload['payload']['success'] = False
  1764. response_payload['payload']['error_code'] = 1008
  1765. response_payload['payload']['error_message'] = f"卡号下发异常: {e}"
  1766. mqtt_client.publish(topic_response_dtu, json.dumps(response_payload), qos=1)
  1767. return
  1768. elif command == 'SYNC_ALL_MAPPING':
  1769. # 批量同步期望映射
  1770. mappings = params.get('mappings', [])
  1771. if not isinstance(mappings, list):
  1772. response_payload['payload']['success'] = False
  1773. response_payload['payload']['error_code'] = 1006
  1774. response_payload['payload']['error_message'] = "mappings 字段格式错误"
  1775. mqtt_client.publish(topic_response_dtu, json.dumps(response_payload), qos=1)
  1776. return
  1777. try:
  1778. for mapping in mappings:
  1779. panel_id = _resolve_panel_id(mapping.get('panel_id'))
  1780. port_id = mapping.get('port_id')
  1781. jumper_uid = mapping.get('jumper_uid')
  1782. # MQTT 16 位 -> 内部 12 位硬件卡号(剥前 4 位 DTU 号),非法抛 ValueError
  1783. expected_uid = _from_mqtt_jumper_uid(jumper_uid)
  1784. if panel_id not in panel_config:
  1785. raise ValueError(f"目标面板不存在: {mapping.get('panel_id')}")
  1786. if panel_id not in port_state:
  1787. port_state[panel_id] = {}
  1788. if port_id not in port_state[panel_id]:
  1789. port_state[panel_id][port_id] = {'last_uid': None, 'expected_uid': None, 'alarm_count': 0}
  1790. port_state[panel_id][port_id]['expected_uid'] = expected_uid
  1791. # 持久化期望映射(内部 12 位)
  1792. dtu_config.setdefault('port_mappings', {}).setdefault(panel_id, {})[str(port_id)] = expected_uid
  1793. save_dtu_config()
  1794. except Exception as e:
  1795. response_payload['payload']['success'] = False
  1796. response_payload['payload']['error_code'] = 1007
  1797. response_payload['payload']['error_message'] = f"同步映射失败: {e}"
  1798. mqtt_client.publish(topic_response_dtu, json.dumps(response_payload), qos=1)
  1799. return
  1800. # V4: 对涉及的面板逐个下发卡号到硬件
  1801. affected_panels = set()
  1802. for mapping in mappings:
  1803. pid = _resolve_panel_id(mapping.get('panel_id'))
  1804. if pid:
  1805. affected_panels.add(pid)
  1806. hw_results = {}
  1807. hw_errors = {}
  1808. for pid in affected_panels:
  1809. device_address = panel_config[pid].get('address', 1)
  1810. card_mappings = _build_card_mappings_for_panel(pid)
  1811. try:
  1812. hw_result = modbus_client.set_card_mapping(device_address, card_mappings, timeout=2.0)
  1813. if 'error' in hw_result:
  1814. hw_errors[pid] = hw_result['error']
  1815. else:
  1816. hw_results[pid] = 'ok'
  1817. logger.info(f"SYNC_ALL_MAPPING 面板{pid} 卡号已下发硬件")
  1818. except Exception as e:
  1819. hw_errors[pid] = str(e)
  1820. if hw_errors:
  1821. response_payload['payload']['success'] = False
  1822. response_payload['payload']['error_code'] = 1008
  1823. response_payload['payload']['error_message'] = f"部分面板卡号下发失败: {hw_errors}"
  1824. response_payload['payload']['result'] = {'hardware': {'success': hw_results, 'errors': hw_errors}}
  1825. elif command == 'REBOOT':
  1826. # 重启DTU: 1s 延迟后执行 reboot
  1827. import subprocess
  1828. logger.warning(f"MQTT REBOOT 命令收到,1 秒后执行系统重启")
  1829. threading.Thread(target=lambda: (
  1830. time.sleep(1),
  1831. subprocess.run(['reboot'], capture_output=True)
  1832. ), daemon=True).start()
  1833. response_payload['payload']['result'] = {'message': 'Reboot scheduled in 1s'}
  1834. elif command == 'READ_PANEL_STATUS':
  1835. # 读取面板状态
  1836. # V4 寄存器: 0x0000=网络通断(保留), 0x0001=LED控制, 0x0002=批量卡号(144字节)
  1837. device_address = 1
  1838. for panel_id, cfg in panel_config.items():
  1839. if panel_id == target:
  1840. device_address = cfg.get('address', 1)
  1841. break
  1842. # V4: 只读 0x0000-0x0001(网络通断 + LED);0x0002 起为批量卡号数据,不再作为状态寄存器
  1843. status_result = modbus_client.read_holding_registers(device_address, 0x0000, 2)
  1844. registers = status_result.get('registers', [])
  1845. # V4: 天线在场状态由批量卡号读得出(24 元素 0/1 列表)
  1846. bulk_result = modbus_client.read_all_antenna_cards(device_address)
  1847. antenna_status = []
  1848. if 'error' not in bulk_result:
  1849. antenna_status = [1 if c.get('present') else 0 for c in bulk_result.get('cards', [])]
  1850. panel_status = {
  1851. 'device_address': device_address,
  1852. 'run_status': registers[0] if len(registers) > 0 else None,
  1853. 'led_control': registers[1] if len(registers) > 1 else None,
  1854. 'antenna_status': antenna_status,
  1855. 'raw_registers': registers
  1856. }
  1857. response_payload['payload']['success'] = 'error' not in status_result
  1858. response_payload['payload']['result'] = panel_status
  1859. elif command == 'QUERY_JUMPER_STATUS':
  1860. # 查询跳线器状态汇总 (7.8) - 触发 STATUS 上行至 jumper/{dtu_id}/status
  1861. dtu_publish_jumper_status()
  1862. return # dtu_publish_jumper_status 已发送响应
  1863. elif command == 'QUERY_ENV_SENSOR':
  1864. # 查询环境传感器 (7.9) - 触发 STATUS 上行至 env/{dtu_id}/sensor
  1865. dtu_publish_env_sensor()
  1866. return # dtu_publish_env_sensor 已发送响应
  1867. elif command == 'OTA_UPGRADE':
  1868. params = payload.get('payload', {}).get('params', {})
  1869. target_version = params.get('firmware_version')
  1870. force_upgrade = params.get('force_upgrade', False)
  1871. current_version = dtu_config.get('firmware_version', 'v1.0.0')
  1872. incoming_msg_id = payload.get('msg_id', '')
  1873. if target_version == current_version and not force_upgrade:
  1874. response_payload['payload']['success'] = False
  1875. response_payload['payload']['error_code'] = 1021
  1876. response_payload['payload']['error_message'] = f"已是目标版本: {target_version}"
  1877. mqtt_client.publish(topic_response_dtu, json.dumps(response_payload), qos=1)
  1878. return
  1879. if params.get('firmware_url'):
  1880. import threading
  1881. threading.Thread(target=lambda: _run_ota({**params, '_incoming_msg_id': incoming_msg_id}), daemon=True).start()
  1882. response_payload['payload']['success'] = True
  1883. response_payload['payload']['result'] = {
  1884. 'firmware_version': target_version,
  1885. 'ota_status': 'DOWNLOADING',
  1886. 'restart_required': False
  1887. }
  1888. elif command == 'OTA_CANCEL':
  1889. # 取消OTA升级
  1890. # 向OTA控制寄存器写入取消命令 (0x0000=取消)
  1891. device_address = 1
  1892. for panel_id, cfg in panel_config.items():
  1893. if panel_id == target:
  1894. device_address = cfg.get('address', 1)
  1895. break
  1896. # OTA控制寄存器地址: 0x0300
  1897. # 值: 0x0000=取消升级
  1898. cancel_result = modbus_client.write_single_register(device_address, 0x0300, 0x0000)
  1899. response_payload['payload']['success'] = 'error' not in cancel_result
  1900. response_payload['payload']['result'] = {
  1901. 'message': 'OTA upgrade cancelled' if 'error' not in cancel_result else 'Failed to cancel OTA',
  1902. 'raw_result': cancel_result
  1903. }
  1904. else:
  1905. # 未知命令
  1906. response_payload['payload']['success'] = False
  1907. response_payload['payload']['error_code'] = 1001
  1908. response_payload['payload']['error_message'] = f"未知命令: {command}"
  1909. mqtt_client.publish(topic_response_dtu, json.dumps(response_payload), qos=1)
  1910. return
  1911. # 发送响应(QUERY_DTU_STATUS 已由 dtu_publish_status 发送;QUERY_JUMPER_STATUS/QUERY_ENV_SENSOR 已在分支中发布)
  1912. if command == 'QUERY_DTU_STATUS':
  1913. pass
  1914. else:
  1915. cmd = command
  1916. if cmd == 'READ_PANEL_STATUS':
  1917. if target and target != 'all' and target in panel_config:
  1918. # 单个面板: 使用 panel_id 作为 topic segment
  1919. topic = build_dtu_topic(dtu_config['customer_id'], 'patchpanel',
  1920. dtu_config['dtu_id'], target, 'status')
  1921. mqtt_client.publish(topic, json.dumps(response_payload), qos=0)
  1922. else:
  1923. # target=all: 遍历所有面板发布到各自的 topic
  1924. for pid in panel_config.keys():
  1925. topic = build_dtu_topic(dtu_config['customer_id'], 'patchpanel',
  1926. dtu_config['dtu_id'], pid, 'status')
  1927. mqtt_client.publish(topic, json.dumps(response_payload), qos=0)
  1928. else:
  1929. topic = build_dtu_topic(dtu_config['customer_id'], 'dtu', dtu_config['dtu_id'], 'response')
  1930. p = response_payload['payload']
  1931. if not p.get('success', True) and p.get('error_code', 0) == 0:
  1932. p['error_code'] = 9999
  1933. p['error_message'] = p.get('error_message') or '未知错误'
  1934. mqtt_client.publish(topic, json.dumps(response_payload), qos=1)
  1935. except Exception as e:
  1936. logger.error(f"处理控制指令失败: {str(e)}")
  1937. # 启动DTU心跳定时器
  1938. def start_dtu_heartbeat():
  1939. """启动DTU心跳定时任务"""
  1940. def heartbeat_task():
  1941. env_tick = 0
  1942. while True:
  1943. socketio.sleep(dtu_config.get('heartbeat_interval', DTU_HEARTBEAT_INTERVAL))
  1944. if _mqtt_connected() and dtu_config.get('enabled'):
  1945. dtu_publish_status()
  1946. # 每 2 个心跳周期 (120s) 主动推送一次环境传感器数据 (7.9)
  1947. env_tick += 1
  1948. if env_tick >= 2:
  1949. env_tick = 0
  1950. dtu_publish_env_sensor()
  1951. socketio.start_background_task(target=heartbeat_task)
  1952. # 修改mqtt_data_handler以处理控制指令
  1953. def mqtt_data_handler_extended(data):
  1954. """处理MQTT接收的数据(扩展版,含DTU协议)"""
  1955. try:
  1956. topic = data.get('topic', '')
  1957. payload_str = data.get('payload', '')
  1958. # 尝试解析JSON
  1959. try:
  1960. payload = json.loads(payload_str) if isinstance(payload_str, str) else payload_str
  1961. except:
  1962. payload = payload_str
  1963. # 添加到缓冲区
  1964. timestamp = time.strftime('%Y-%m-%d %H:%M:%S')
  1965. mqtt_data_buffer.append({
  1966. 'timestamp': timestamp,
  1967. 'topic': topic,
  1968. 'payload': payload_str
  1969. })
  1970. if len(mqtt_data_buffer) > MAX_BUFFER_SIZE:
  1971. mqtt_data_buffer.pop(0)
  1972. # 通过WebSocket广播
  1973. socketio.emit('mqtt_data', {
  1974. 'timestamp': timestamp,
  1975. 'topic': topic,
  1976. 'payload': payload_str
  1977. }, namespace=SOCKETIO_NAMESPACE_DATA)
  1978. # 检查是否是控制指令主题
  1979. expected_control_topic = build_dtu_topic(dtu_config['customer_id'], 'dtu', dtu_config['dtu_id'], 'control')
  1980. if topic == expected_control_topic:
  1981. dtu_handle_control(topic, payload)
  1982. # 检查是否是状态主题,用于处理OTA状态上报
  1983. expected_status_topic = build_dtu_topic(dtu_config['customer_id'], 'dtu', dtu_config['dtu_id'], 'status')
  1984. if topic == expected_status_topic and isinstance(payload, dict):
  1985. if 'ota_status' in payload or 'firmware_version' in payload:
  1986. handle_ota_status(dtu_config['dtu_id'], payload)
  1987. # 检查是否是环境传感器数据主题 (兼容旧 sensor/.../data 路径,spec 7.9 为 env/.../sensor)
  1988. sensor_topic = build_dtu_topic(dtu_config['customer_id'], 'sensor', dtu_config['dtu_id'], 'data')
  1989. env_topic = build_dtu_topic(dtu_config['customer_id'], 'env', dtu_config['dtu_id'], 'sensor')
  1990. if topic in (sensor_topic, env_topic) and isinstance(payload, dict):
  1991. if 'temperature' in payload or 'humidity' in payload:
  1992. update_env_sensor_data(
  1993. payload.get('temperature'),
  1994. payload.get('humidity'),
  1995. payload.get('dtu_temperature'),
  1996. payload.get('sensor_update_time')
  1997. )
  1998. # 检查是否是广播主题 - DTU发现 (7.14)
  1999. discover_topic = build_dtu_topic('broadcast', 'dtu', 'discover')
  2000. if topic == discover_topic:
  2001. handle_broadcast_discover(payload)
  2002. # 检查是否是广播主题 - 批量配置 (7.15)
  2003. config_topic = build_dtu_topic('broadcast', 'dtu', 'config')
  2004. if topic == config_topic:
  2005. handle_broadcast_config(payload)
  2006. # 转发到串口(如果启用)
  2007. if forward_mqtt_to_serial and _serial_connected():
  2008. success, msg = serial_client.send_data(payload_str)
  2009. if not success:
  2010. logger.warning(f"MQTT数据转发到串口失败: {msg}")
  2011. except Exception as e:
  2012. logger.error(f"处理MQTT数据时出错: {str(e)}")
  2013. # WebSocket事件处理
  2014. @socketio.on('connect', namespace=SOCKETIO_NAMESPACE_DATA)
  2015. def handle_data_connect():
  2016. """处理数据命名空间的连接"""
  2017. try:
  2018. client_id = str(uuid.uuid4())
  2019. connected_clients['data'].add(client_id)
  2020. logger.info(f'客户端已连接到数据命名空间,当前连接数: {len(connected_clients["data"])}')
  2021. # 发送当前的缓冲区数据,限制发送的历史记录数量
  2022. max_history = 100 # 限制发送的历史记录数量以提高性能
  2023. emit('serial_data_history', {'data': serial_data_buffer[-max_history:]})
  2024. emit('mqtt_data_history', {'data': mqtt_data_buffer[-max_history:]})
  2025. # 存储客户端ID以便断开连接时使用
  2026. socketio.start_background_task(target=lambda: None) # 确保上下文可用
  2027. except Exception as e:
  2028. logger.error(f"处理数据命名空间连接时出错: {str(e)}")
  2029. @socketio.on('connect', namespace=SOCKETIO_NAMESPACE_STATUS)
  2030. def handle_status_connect():
  2031. """处理状态命名空间的连接"""
  2032. try:
  2033. client_id = str(uuid.uuid4())
  2034. connected_clients['status'].add(client_id)
  2035. logger.info(f'客户端已连接到状态命名空间,当前连接数: {len(connected_clients["status"])}')
  2036. # 发送当前状态
  2037. emit('serial_status', {'connected': serial_status})
  2038. emit('mqtt_status', {'connected': mqtt_status})
  2039. emit('forward_status', {
  2040. 'serial_to_mqtt': forward_serial_to_mqtt,
  2041. 'mqtt_to_serial': forward_mqtt_to_serial,
  2042. 'publish_topic': mqtt_publish_topic
  2043. })
  2044. except Exception as e:
  2045. logger.error(f"处理状态命名空间连接时出错: {str(e)}")
  2046. @socketio.on('connect', namespace=SOCKETIO_NAMESPACE_CONTROL)
  2047. def handle_control_connect():
  2048. """处理控制命名空间的连接"""
  2049. try:
  2050. client_id = str(uuid.uuid4())
  2051. connected_clients['control'].add(client_id)
  2052. logger.info(f'客户端已连接到控制命名空间,当前连接数: {len(connected_clients["control"])}')
  2053. except Exception as e:
  2054. logger.error(f"处理控制命名空间连接时出错: {str(e)}")
  2055. @socketio.on('disconnect', namespace=SOCKETIO_NAMESPACE_DATA)
  2056. def handle_data_disconnect():
  2057. """处理数据命名空间的断开连接"""
  2058. try:
  2059. # 清理客户端连接记录
  2060. # 在实际应用中,可能需要更复杂的逻辑来追踪具体哪个客户端断开了连接
  2061. if len(connected_clients['data']) > 0:
  2062. # 这里简化处理,实际应该维护session到client_id的映射
  2063. connected_clients['data'].pop() # 注意:这是一个简化的实现
  2064. logger.info(f'客户端已断开数据命名空间的连接,当前连接数: {len(connected_clients["data"])}')
  2065. except Exception as e:
  2066. logger.error(f"处理数据命名空间断开连接时出错: {str(e)}")
  2067. @socketio.on('disconnect', namespace=SOCKETIO_NAMESPACE_STATUS)
  2068. def handle_status_disconnect():
  2069. """处理状态命名空间的断开连接"""
  2070. try:
  2071. # 清理客户端连接记录
  2072. if len(connected_clients['status']) > 0:
  2073. connected_clients['status'].pop()
  2074. logger.info(f'客户端已断开状态命名空间的连接,当前连接数: {len(connected_clients["status"])}')
  2075. except Exception as e:
  2076. logger.error(f"处理状态命名空间断开连接时出错: {str(e)}")
  2077. @socketio.on('disconnect', namespace=SOCKETIO_NAMESPACE_CONTROL)
  2078. def handle_control_disconnect():
  2079. """处理控制命名空间的断开连接"""
  2080. try:
  2081. # 清理客户端连接记录
  2082. if len(connected_clients['control']) > 0:
  2083. connected_clients['control'].pop()
  2084. logger.info(f'客户端已断开控制命名空间的连接,当前连接数: {len(connected_clients["control"])}')
  2085. except Exception as e:
  2086. logger.error(f"处理控制命名空间断开连接时出错: {str(e)}")
  2087. @socketio.on('serial_send', namespace=SOCKETIO_NAMESPACE_CONTROL)
  2088. def handle_serial_send(data):
  2089. """通过WebSocket处理串口发送数据请求"""
  2090. try:
  2091. message = data.get('message', '')
  2092. if not message:
  2093. emit('serial_send_response', {
  2094. 'success': False,
  2095. 'message': '消息内容不能为空',
  2096. 'error_code': ERROR_CODES['CONFIG_ERROR']
  2097. })
  2098. return
  2099. if not (isinstance(serial_client.get_status(), dict) and serial_client.get_status().get("connected", False)):
  2100. emit('serial_send_response', {
  2101. 'success': False,
  2102. 'message': '串口未连接',
  2103. 'error_code': ERROR_CODES['SERIAL_CONNECTION_ERROR']
  2104. })
  2105. return
  2106. success, msg = serial_client.send_data(message)
  2107. emit('serial_send_response', {
  2108. 'success': success,
  2109. 'message': msg,
  2110. 'error_code': ERROR_CODES['SUCCESS'] if success else ERROR_CODES['SERIAL_SEND_ERROR']
  2111. })
  2112. if success:
  2113. logger.info(f"通过WebSocket发送串口数据成功: {message[:50]}..." if len(message) > 50 else message)
  2114. except Exception as e:
  2115. error_msg = f"处理串口发送请求时出错: {str(e)}"
  2116. logger.error(error_msg)
  2117. emit('serial_send_response', {
  2118. 'success': False,
  2119. 'message': error_msg,
  2120. 'error_code': ERROR_CODES['UNKNOWN_ERROR']
  2121. })
  2122. @socketio.on('mqtt_publish', namespace=SOCKETIO_NAMESPACE_CONTROL)
  2123. def handle_mqtt_publish(data):
  2124. """通过WebSocket处理MQTT发布数据请求"""
  2125. try:
  2126. topic = data.get('topic', '')
  2127. message = data.get('message', '')
  2128. if not topic or not message:
  2129. emit('mqtt_publish_response', {
  2130. 'success': False,
  2131. 'message': '主题或消息内容不能为空',
  2132. 'error_code': ERROR_CODES['CONFIG_ERROR']
  2133. })
  2134. return
  2135. if not _mqtt_connected():
  2136. emit('mqtt_publish_response', {
  2137. 'success': False,
  2138. 'message': 'MQTT未连接',
  2139. 'error_code': ERROR_CODES['MQTT_CONNECTION_ERROR']
  2140. })
  2141. return
  2142. success, msg = mqtt_client.publish(topic, message)
  2143. emit('mqtt_publish_response', {
  2144. 'success': success,
  2145. 'message': msg,
  2146. 'error_code': ERROR_CODES['SUCCESS'] if success else ERROR_CODES['MQTT_PUBLISH_ERROR']
  2147. })
  2148. if success:
  2149. logger.info(f"通过WebSocket发布MQTT消息成功: 主题={topic}, 消息={message[:50]}..." if len(message) > 50 else message)
  2150. except Exception as e:
  2151. error_msg = f"处理MQTT发布请求时出错: {str(e)}"
  2152. logger.error(error_msg)
  2153. emit('mqtt_publish_response', {
  2154. 'success': False,
  2155. 'message': error_msg,
  2156. 'error_code': ERROR_CODES['UNKNOWN_ERROR']
  2157. })
  2158. @socketio.on('update_forward_config', namespace=SOCKETIO_NAMESPACE_CONTROL)
  2159. def handle_update_forward_config(data):
  2160. """通过WebSocket更新转发配置"""
  2161. global forward_serial_to_mqtt, forward_mqtt_to_serial, mqtt_publish_topic
  2162. try:
  2163. # 更新转发标志
  2164. if 'serial_to_mqtt' in data:
  2165. forward_serial_to_mqtt = bool(data['serial_to_mqtt'])
  2166. if 'mqtt_to_serial' in data:
  2167. forward_mqtt_to_serial = bool(data['mqtt_to_serial'])
  2168. if 'publish_topic' in data:
  2169. new_topic = str(data['publish_topic'])
  2170. if not new_topic.strip():
  2171. raise ValueError("发布主题不能为空")
  2172. mqtt_publish_topic = new_topic
  2173. # 广播配置更新
  2174. socketio.emit('forward_status', {
  2175. 'serial_to_mqtt': forward_serial_to_mqtt,
  2176. 'mqtt_to_serial': forward_mqtt_to_serial,
  2177. 'publish_topic': mqtt_publish_topic
  2178. }, namespace=SOCKETIO_NAMESPACE_STATUS)
  2179. emit('update_forward_config_response', {
  2180. 'success': True,
  2181. 'message': '转发配置已更新',
  2182. 'error_code': ERROR_CODES['SUCCESS']
  2183. })
  2184. logger.info(f"转发配置已更新 - 串口到MQTT: {forward_serial_to_mqtt}, MQTT到串口: {forward_mqtt_to_serial}, 发布主题: {mqtt_publish_topic}")
  2185. except ValueError as e:
  2186. error_msg = str(e)
  2187. logger.warning(f"转发配置更新失败: {error_msg}")
  2188. emit('update_forward_config_response', {
  2189. 'success': False,
  2190. 'message': error_msg,
  2191. 'error_code': ERROR_CODES['CONFIG_ERROR']
  2192. })
  2193. except Exception as e:
  2194. error_msg = f'更新转发配置失败: {str(e)}'
  2195. logger.error(error_msg)
  2196. emit('update_forward_config_response', {
  2197. 'success': False,
  2198. 'message': error_msg,
  2199. 'error_code': ERROR_CODES['UNKNOWN_ERROR']
  2200. })
  2201. @socketio.on('clear_data_buffer', namespace=SOCKETIO_NAMESPACE_CONTROL)
  2202. def handle_clear_data_buffer(data):
  2203. """通过WebSocket清空数据缓冲区"""
  2204. try:
  2205. buffer_type = data.get('type', '')
  2206. if buffer_type == 'serial':
  2207. serial_data_buffer.clear()
  2208. emit('clear_data_buffer_response', {
  2209. 'success': True,
  2210. 'message': '串口数据缓冲区已清空',
  2211. 'error_code': ERROR_CODES['SUCCESS']
  2212. })
  2213. logger.info("串口数据缓冲区已清空")
  2214. elif buffer_type == 'mqtt':
  2215. mqtt_data_buffer.clear()
  2216. emit('clear_data_buffer_response', {
  2217. 'success': True,
  2218. 'message': 'MQTT数据缓冲区已清空',
  2219. 'error_code': ERROR_CODES['SUCCESS']
  2220. })
  2221. logger.info("MQTT数据缓冲区已清空")
  2222. elif buffer_type == 'all':
  2223. serial_data_buffer.clear()
  2224. mqtt_data_buffer.clear()
  2225. emit('clear_data_buffer_response', {
  2226. 'success': True,
  2227. 'message': '所有数据缓冲区已清空',
  2228. 'error_code': ERROR_CODES['SUCCESS']
  2229. })
  2230. logger.info("所有数据缓冲区已清空")
  2231. else:
  2232. emit('clear_data_buffer_response', {
  2233. 'success': False,
  2234. 'message': '无效的缓冲区类型',
  2235. 'error_code': ERROR_CODES['CONFIG_ERROR']
  2236. })
  2237. logger.warning(f"清空缓冲区失败: 无效的缓冲区类型 '{buffer_type}'")
  2238. except Exception as e:
  2239. error_msg = f'清空缓冲区失败: {str(e)}'
  2240. logger.error(error_msg)
  2241. emit('clear_data_buffer_response', {
  2242. 'success': False,
  2243. 'message': error_msg,
  2244. 'error_code': ERROR_CODES['UNKNOWN_ERROR']
  2245. })
  2246. # 设置回调
  2247. serial_client.set_data_callback(serial_data_handler)
  2248. serial_client.set_send_callback(serial_send_handler)
  2249. serial_client.set_status_callback(serial_status_handler)
  2250. mqtt_client.set_data_callback(mqtt_data_handler_extended)
  2251. mqtt_client.set_status_callback(mqtt_status_handler)
  2252. # 设备配置文件路径
  2253. DEVICE_CONFIG_FILE = '/root/dzxj_dtu/devices.json'
  2254. confirm_loop_running = False
  2255. def save_device_config():
  2256. """保存设备配置到文件(快照 + 原子写)"""
  2257. try:
  2258. filepath = DEVICE_CONFIG_FILE
  2259. success = _save_config_file(filepath, address_config.get_stored_devices(), '设备配置')
  2260. if success:
  2261. logger.info(f"设备配置已保存到: {filepath}")
  2262. except Exception as e:
  2263. logger.error(f"保存设备配置失败: {str(e)}")
  2264. def load_device_config():
  2265. """加载设备配置(文件损坏时回退最新快照)"""
  2266. try:
  2267. filepath = DEVICE_CONFIG_FILE
  2268. if os.path.exists(filepath):
  2269. try:
  2270. devices = config_backup.read_json_file(filepath)
  2271. except Exception as e:
  2272. logger.error(f"读取设备配置失败: {e},尝试回退快照")
  2273. devices, snap = config_backup.latest_snapshot_content(filepath, BACKUP_DIR)
  2274. if not isinstance(devices, dict):
  2275. return {}
  2276. logger.warning(f"已从快照恢复设备配置: {snap}")
  2277. if isinstance(devices, dict):
  2278. address_config.stored_devices = devices
  2279. logger.info(f"已加载 {len(devices)} 个设备配置")
  2280. return devices
  2281. except Exception as e:
  2282. logger.error(f"加载设备配置失败: {str(e)}")
  2283. return {}
  2284. def _merge_panel_config(new_config):
  2285. """合并面板配置:叠加新发现的在线设备,并保留已标记为离线的面板
  2286. 离线设备保留在 panel_config 中(status='offline'),UI 可显示离线状态,
  2287. 轮询循环会跳过它们不再发查询;广播发现仍会定期尝试重新上线。
  2288. """
  2289. global panel_config
  2290. merged = dict(new_config)
  2291. for panel_id, cfg in panel_config.items():
  2292. if cfg.get('status') == 'offline' and panel_id not in merged:
  2293. merged[panel_id] = cfg
  2294. panel_config = merged
  2295. return merged
  2296. def sync_panel_config():
  2297. """根据 address_config.stored_devices 同步 panel_config"""
  2298. global panel_config
  2299. try:
  2300. devices = address_config.get_stored_devices()
  2301. panel_id_map = dtu_config.get('panel_id_mapping', {}) or {}
  2302. new_config = {}
  2303. for uid_hex, addr in devices.items():
  2304. custom_id = panel_id_map.get(uid_hex) or panel_id_map.get(addr)
  2305. if custom_id:
  2306. panel_id = str(custom_id)
  2307. else:
  2308. panel_id = f"PANEL_{dtu_config.get('dtu_id', 'DTU001')}_{addr}"
  2309. new_config[panel_id] = {
  2310. 'address': addr,
  2311. 'position': addr,
  2312. 'panel_uid': uid_hex,
  2313. 'status': 'online'
  2314. }
  2315. _merge_panel_config(new_config)
  2316. logger.info(f"已同步面板配置: {len(panel_config)} 个面板")
  2317. except Exception as e:
  2318. logger.error(f"同步面板配置失败: {str(e)}")
  2319. def sync_panel_config_from_results(results):
  2320. """根据本次自动配置结果同步 panel_config,只把真实发现/分配到的设备加入轮询"""
  2321. global panel_config
  2322. try:
  2323. panel_id_map = dtu_config.get('panel_id_mapping', {}) or {}
  2324. new_config = {}
  2325. for r in results:
  2326. if r.get('status') != 'assigned':
  2327. continue
  2328. uid_hex = r.get('uid')
  2329. addr = r.get('address')
  2330. if not uid_hex or not addr:
  2331. continue
  2332. custom_id = panel_id_map.get(uid_hex) or panel_id_map.get(addr)
  2333. if custom_id:
  2334. panel_id = str(custom_id)
  2335. else:
  2336. panel_id = f"PANEL_{dtu_config.get('dtu_id', 'DTU001')}_{addr}"
  2337. new_config[panel_id] = {
  2338. 'address': addr,
  2339. 'position': addr,
  2340. 'panel_uid': uid_hex,
  2341. 'status': 'online'
  2342. }
  2343. _merge_panel_config(new_config)
  2344. logger.info(f"已同步面板配置(发现模式): {len(panel_config)} 个面板")
  2345. except Exception as e:
  2346. logger.error(f"同步面板配置失败: {str(e)}")
  2347. def confirm_loop():
  2348. """后台确认线程:每10秒对所有已存储设备发送 confirm_address"""
  2349. global confirm_loop_running
  2350. confirm_loop_running = True
  2351. logger.info("启动设备确认线程 (间隔10秒)")
  2352. while confirm_loop_running:
  2353. try:
  2354. devices = address_config.get_stored_devices()
  2355. if not devices:
  2356. time.sleep(10)
  2357. continue
  2358. if not _serial_connected():
  2359. time.sleep(10)
  2360. continue
  2361. _st = serial_client.get_status()
  2362. if not (isinstance(_st, dict) and _st.get('connected', False)):
  2363. time.sleep(10)
  2364. continue
  2365. for uid_hex, addr in devices.items():
  2366. try:
  2367. uid_bytes = bytes.fromhex(uid_hex)
  2368. cmd = build_confirm_address(addr, uid_bytes)
  2369. success, msg = serial_client.send_raw(cmd)
  2370. if success:
  2371. logger.debug(f"确认设备: 地址={addr}, UID={uid_hex[:16]}...")
  2372. else:
  2373. logger.warning(f"确认失败 地址={addr}: {msg}")
  2374. except Exception as e:
  2375. logger.error(f"确认异常 地址={addr}: {str(e)}")
  2376. time.sleep(0.1)
  2377. except Exception as e:
  2378. logger.error(f"确认线程异常: {str(e)}")
  2379. time.sleep(10)
  2380. # API路由
  2381. # 移除静态文件服务,前端由nginx提供服务
  2382. # 根路径路由
  2383. @app.route('/')
  2384. def index():
  2385. return send_from_directory(STATIC_FOLDER, 'index.html')
  2386. # 404错误处理器 - 解决SPA路由刷新问题
  2387. @app.errorhandler(404)
  2388. def page_not_found(e):
  2389. """处理所有404错误,对于非API路径返回index.html"""
  2390. path = request.path
  2391. # 检查是否是API请求
  2392. if path.startswith('/api/'):
  2393. # 对于API请求,返回404错误
  2394. return jsonify({
  2395. 'success': False,
  2396. 'message': 'API endpoint not found'
  2397. }), 404
  2398. # 对于所有非API路径,返回index.html让前端路由处理
  2399. return send_from_directory(STATIC_FOLDER, 'index.html'), 200
  2400. @app.route('/api/serial/ports', methods=['GET'])
  2401. def get_serial_ports():
  2402. """获取可用串口列表"""
  2403. ports = serial_client.list_ports()
  2404. return jsonify({
  2405. 'success': True,
  2406. 'ports': ports
  2407. })
  2408. @app.route('/api/serial/connect', methods=['POST'])
  2409. def serial_connect():
  2410. """连接串口"""
  2411. try:
  2412. data = request.json
  2413. port = data.get('port')
  2414. baudrate = data.get('baudrate', 9600)
  2415. bytesize = data.get('bytesize', 8)
  2416. parity = data.get('parity', 'N')
  2417. stopbits = data.get('stopbits', 1)
  2418. timeout = data.get('timeout', 0.1)
  2419. if not port:
  2420. logger.warning("连接串口请求缺少串口名称")
  2421. return jsonify({
  2422. 'success': False,
  2423. 'message': '串口名称不能为空',
  2424. 'error_code': ERROR_CODES['CONFIG_ERROR']
  2425. }), 400
  2426. # 先断开之前的连接
  2427. _status = serial_client.get_status()
  2428. if isinstance(_status, dict) and _status.get('connected', False):
  2429. _port = serial_client.current_config.port if serial_client.current_config else 'unknown'
  2430. logger.info(f"断开现有串口连接: {_port}")
  2431. serial_client.disconnect()
  2432. # 连接新的串口
  2433. logger.info(f"尝试连接串口: {port}, 波特率: {baudrate}")
  2434. success, message = serial_client.connect(port, baudrate=baudrate, timeout=timeout, bytesize=bytesize, parity=parity, stopbits=stopbits)
  2435. status_code = 200 if success else 400
  2436. error_code = ERROR_CODES['SUCCESS'] if success else ERROR_CODES['SERIAL_CONNECTION_ERROR']
  2437. response = {
  2438. 'success': success,
  2439. 'message': message,
  2440. 'error_code': error_code
  2441. }
  2442. if success:
  2443. logger.info(f"串口连接成功: {port}")
  2444. # 保存串口配置
  2445. save_serial_config(port, baudrate, timeout, bytesize, parity, stopbits)
  2446. else:
  2447. logger.error(f"串口连接失败: {message}")
  2448. return jsonify(response), status_code
  2449. except Exception as e:
  2450. error_msg = f'连接串口时出错: {str(e)}'
  2451. logger.exception(error_msg) # 使用exception记录完整堆栈
  2452. return jsonify({
  2453. 'success': False,
  2454. 'message': error_msg,
  2455. 'error_code': ERROR_CODES['UNKNOWN_ERROR']
  2456. }), 500
  2457. @app.route('/api/serial/disconnect', methods=['POST'])
  2458. def serial_disconnect():
  2459. """断开串口连接"""
  2460. success, message = serial_client.disconnect()
  2461. return jsonify({
  2462. 'success': success,
  2463. 'message': message
  2464. })
  2465. @app.route('/api/serial/status', methods=['GET'])
  2466. def serial_get_status():
  2467. """获取串口状态"""
  2468. _st = serial_client.get_status()
  2469. saved_config = {}
  2470. try:
  2471. with open(SERIAL_CONFIG_FILE, 'r') as f:
  2472. saved_config = json.load(f)
  2473. except (FileNotFoundError, json.JSONDecodeError):
  2474. pass
  2475. result = {'saved_config': saved_config}
  2476. if isinstance(_st, dict):
  2477. result['connected'] = _st.get('connected', False)
  2478. if result['connected']:
  2479. cfg = _st.get('config')
  2480. result['port'] = cfg.port if cfg else None
  2481. else:
  2482. result['port'] = None
  2483. else:
  2484. result['connected'] = bool(_st)
  2485. result['port'] = None
  2486. return jsonify(result)
  2487. @app.route('/api/serial/send', methods=['POST'])
  2488. def serial_send():
  2489. """发送数据到串口"""
  2490. data = request.json
  2491. message = data.get('message')
  2492. if not message:
  2493. return jsonify({
  2494. 'success': False,
  2495. 'message': '消息内容不能为空'
  2496. })
  2497. success, message = serial_client.send_data(message)
  2498. return jsonify({
  2499. 'success': success,
  2500. 'message': message
  2501. })
  2502. @app.route('/api/mqtt/connect', methods=['POST'])
  2503. def mqtt_connect():
  2504. """连接MQTT服务器"""
  2505. try:
  2506. data = request.json
  2507. host = data.get('broker') or data.get('host', 'localhost')
  2508. port = data.get('port', 1883)
  2509. client_id = data.get('client_id', f'serial_gateway_{int(time.time())}')
  2510. username = data.get('username')
  2511. password = data.get('password')
  2512. keepalive = data.get('keepalive', 60)
  2513. # 先断开之前的连接
  2514. if _mqtt_connected():
  2515. _old = mqtt_client.config
  2516. _old_info = f"{_old.broker}:{_old.port}" if _old else "unknown"
  2517. logger.info(f"断开现有MQTT连接: {_old_info}")
  2518. mqtt_client.disconnect()
  2519. # 连接新的MQTT服务器
  2520. logger.info(f"尝试连接MQTT服务器: {host}:{port}, 客户端ID: {client_id}")
  2521. will_topic = build_dtu_topic(dtu_config['customer_id'], 'dtu', dtu_config['dtu_id'], 'status')
  2522. will_payload = json.dumps({
  2523. 'dtu_id': dtu_config['dtu_id'], 'type': 'STATUS',
  2524. 'payload': {'online': False, 'reason': 'CONNECTION_LOST'}
  2525. })
  2526. success, message = mqtt_client.connect(
  2527. broker=host, port=port, client_id=client_id,
  2528. username=username or "", password=password or "",
  2529. keepalive=keepalive,
  2530. will_topic=will_topic, will_payload=will_payload, will_qos=1, will_retain=False
  2531. )
  2532. # 如果连接成功,订阅主题
  2533. if success and 'topics' in data:
  2534. mqtt_client.subscribe(data['topics'])
  2535. status_code = 200 if success else 400
  2536. error_code = ERROR_CODES['SUCCESS'] if success else ERROR_CODES['MQTT_CONNECTION_ERROR']
  2537. response = {
  2538. 'success': success,
  2539. 'message': message,
  2540. 'error_code': error_code
  2541. }
  2542. if success:
  2543. logger.info(f"MQTT服务器连接成功: {host}:{port}")
  2544. save_mqtt_config(
  2545. broker=host, port=port, client_id=client_id,
  2546. username=username or "", password=password or "",
  2547. keepalive=keepalive,
  2548. topics=data.get('topics', [])
  2549. )
  2550. else:
  2551. logger.error(f"MQTT服务器连接失败: {message}")
  2552. return jsonify(response), status_code
  2553. except Exception as e:
  2554. error_msg = f'连接MQTT服务器时出错: {str(e)}'
  2555. logger.exception(error_msg)
  2556. return jsonify({
  2557. 'success': False,
  2558. 'message': error_msg,
  2559. 'error_code': ERROR_CODES['UNKNOWN_ERROR']
  2560. }), 500
  2561. @app.route('/api/mqtt/disconnect', methods=['POST'])
  2562. def mqtt_disconnect():
  2563. """断开MQTT连接"""
  2564. success, message = mqtt_client.disconnect()
  2565. return jsonify({
  2566. 'success': success,
  2567. 'message': message
  2568. })
  2569. @app.route('/api/mqtt/status', methods=['GET'])
  2570. def mqtt_get_status():
  2571. """获取MQTT状态"""
  2572. _st = mqtt_client.get_status()
  2573. return jsonify({
  2574. 'connected': _st.get('connected', False) if isinstance(_st, dict) else bool(_st)
  2575. })
  2576. @app.route('/api/mqtt/publish', methods=['POST'])
  2577. def mqtt_publish():
  2578. """发布MQTT消息"""
  2579. data = request.json
  2580. topic = data.get('topic')
  2581. message = data.get('message')
  2582. if not topic or not message:
  2583. return jsonify({
  2584. 'success': False,
  2585. 'message': '主题和消息内容不能为空'
  2586. })
  2587. success, message = mqtt_client.publish(topic, message)
  2588. return jsonify({
  2589. 'success': success,
  2590. 'message': message
  2591. })
  2592. @app.route('/api/mqtt/subscribe', methods=['POST'])
  2593. def mqtt_subscribe():
  2594. """订阅MQTT主题"""
  2595. data = request.json
  2596. topics = data.get('topics', [])
  2597. if not topics:
  2598. return jsonify({
  2599. 'success': False,
  2600. 'message': '请至少订阅一个主题'
  2601. })
  2602. success, message = mqtt_client.subscribe(topics)
  2603. return jsonify({
  2604. 'success': success,
  2605. 'message': message
  2606. })
  2607. @app.route('/api/mqtt/broadcast_discover', methods=['POST'])
  2608. def mqtt_broadcast_discover():
  2609. """发送DTU发现广播 (7.14)
  2610. 服务器通过广播主题发送DISCOVER消息,
  2611. 触发所有在线DTU进行注册响应
  2612. """
  2613. data = request.json or {}
  2614. customer_id = data.get('customer_id', dtu_config.get('customer_id', 'default'))
  2615. discover_topic = build_dtu_topic('broadcast', 'dtu', 'discover')
  2616. request_id = f"dsc_{int(time.time() * 1000)}"
  2617. payload = {
  2618. 'msg_id': request_id,
  2619. 'timestamp': int(time.time() * 1000),
  2620. 'dtu_id': 'PLATFORM',
  2621. 'type': 'DISCOVER',
  2622. 'payload': {
  2623. 'request_id': request_id,
  2624. 'customer_id': customer_id
  2625. }
  2626. }
  2627. success, message = mqtt_client.publish(discover_topic, json.dumps(payload), qos=1)
  2628. logger.info(f"发送DTU发现广播: topic={discover_topic}, payload={payload}")
  2629. return jsonify({
  2630. 'success': success,
  2631. 'message': 'DTU发现广播已发送' if success else message,
  2632. 'topic': discover_topic
  2633. })
  2634. @app.route('/api/mqtt/broadcast_config', methods=['POST'])
  2635. def mqtt_broadcast_config():
  2636. """发送DTU批量配置广播 (7.15)
  2637. 服务器通过广播主题发送CONFIG消息,
  2638. 用于批量配置所有在线DTU
  2639. """
  2640. data = request.json or {}
  2641. customer_id = data.get('customer_id', dtu_config.get('customer_id', 'default'))
  2642. config_items = data.get('items', {})
  2643. config_version = data.get('config_version', f"cfg_{int(time.time())}")
  2644. force_apply = data.get('force_apply', False)
  2645. config_topic = build_dtu_topic('broadcast', 'dtu', 'config')
  2646. payload = {
  2647. 'msg_id': f"cfg_{int(time.time() * 1000)}",
  2648. 'timestamp': int(time.time() * 1000),
  2649. 'dtu_id': 'PLATFORM',
  2650. 'type': 'CONFIG',
  2651. 'payload': {
  2652. 'customer_id': customer_id,
  2653. 'config_version': config_version,
  2654. 'force_apply': force_apply,
  2655. 'items': config_items
  2656. }
  2657. }
  2658. success, message = mqtt_client.publish(config_topic, json.dumps(payload), qos=1)
  2659. logger.info(f"发送DTU批量配置广播: topic={config_topic}, payload={payload}")
  2660. return jsonify({
  2661. 'success': success,
  2662. 'message': 'DTU批量配置广播已发送' if success else message,
  2663. 'topic': config_topic,
  2664. 'config': {
  2665. 'config_version': config_version,
  2666. 'items': config_items,
  2667. 'force_apply': force_apply
  2668. }
  2669. })
  2670. @app.route('/api/data/serial', methods=['GET'])
  2671. def get_serial_data():
  2672. """获取串口数据"""
  2673. return jsonify({
  2674. 'data': serial_data_buffer
  2675. })
  2676. @app.route('/api/data/mqtt', methods=['GET'])
  2677. def get_mqtt_data():
  2678. """获取MQTT数据"""
  2679. return jsonify({
  2680. 'data': mqtt_data_buffer
  2681. })
  2682. @app.route('/api/forward/config', methods=['POST'])
  2683. def set_forward_config():
  2684. """设置转发配置"""
  2685. global forward_serial_to_mqtt, forward_mqtt_to_serial, mqtt_publish_topic
  2686. data = request.json
  2687. forward_serial_to_mqtt = data.get('serial_to_mqtt', False)
  2688. forward_mqtt_to_serial = data.get('mqtt_to_serial', False)
  2689. if 'publish_topic' in data:
  2690. mqtt_publish_topic = data['publish_topic']
  2691. return jsonify({
  2692. 'success': True,
  2693. 'message': '转发配置已更新',
  2694. 'config': {
  2695. 'serial_to_mqtt': forward_serial_to_mqtt,
  2696. 'mqtt_to_serial': forward_mqtt_to_serial,
  2697. 'publish_topic': mqtt_publish_topic
  2698. }
  2699. })
  2700. @app.route('/api/forward/status', methods=['GET'])
  2701. def get_forward_status():
  2702. """获取转发状态"""
  2703. return jsonify({
  2704. 'serial_to_mqtt': forward_serial_to_mqtt,
  2705. 'mqtt_to_serial': forward_mqtt_to_serial,
  2706. 'publish_topic': mqtt_publish_topic
  2707. })
  2708. # 健康检查端点
  2709. @app.route('/api/health', methods=['GET'])
  2710. def health_check():
  2711. """健康检查端点"""
  2712. try:
  2713. # 获取客户端连接数量
  2714. client_counts = {
  2715. 'data': len(connected_clients['data']),
  2716. 'status': len(connected_clients['status']),
  2717. 'control': len(connected_clients['control'])
  2718. }
  2719. # 获取缓冲区大小
  2720. buffer_sizes = {
  2721. 'serial': len(serial_data_buffer),
  2722. 'mqtt': len(mqtt_data_buffer)
  2723. }
  2724. # 执行系统负载检查
  2725. # 注意:这只是一个简化的负载检查,实际应用中可能需要更复杂的监控
  2726. is_healthy = True
  2727. load_warnings = []
  2728. # 检查缓冲区是否过大
  2729. if buffer_sizes['serial'] > MAX_BUFFER_SIZE * 0.8:
  2730. is_healthy = False
  2731. load_warnings.append(f"串口缓冲区接近最大容量: {buffer_sizes['serial']}/{MAX_BUFFER_SIZE}")
  2732. if buffer_sizes['mqtt'] > MAX_BUFFER_SIZE * 0.8:
  2733. is_healthy = False
  2734. load_warnings.append(f"MQTT缓冲区接近最大容量: {buffer_sizes['mqtt']}/{MAX_BUFFER_SIZE}")
  2735. # 检查WebSocket连接数是否过多
  2736. total_clients = sum(client_counts.values())
  2737. if total_clients > 100: # 设置合理的阈值
  2738. is_healthy = False
  2739. load_warnings.append(f"WebSocket连接数过多: {total_clients}")
  2740. # 尝试获取网络状态信息(使用try-except包装,防止网络模块出错导致健康检查失败)
  2741. network_info = None
  2742. try:
  2743. network_info = network_manager.get_network_status()
  2744. except Exception as e:
  2745. logger.warning(f"获取网络状态时出错: {str(e)}")
  2746. # 不影响整体健康检查,只添加警告
  2747. load_warnings.append(f"网络状态获取失败: {str(e)}")
  2748. response = {
  2749. 'status': 'healthy' if is_healthy else 'warning',
  2750. 'timestamp': time.strftime('%Y-%m-%d %H:%M:%S'),
  2751. 'services': {
  2752. 'serial': serial_status,
  2753. 'mqtt': mqtt_status,
  2754. 'websocket': True
  2755. },
  2756. 'client_counts': client_counts,
  2757. 'buffer_sizes': buffer_sizes,
  2758. 'warnings': load_warnings
  2759. }
  2760. # 仅当获取到网络信息时添加
  2761. if network_info:
  2762. response['network'] = network_info
  2763. return jsonify(response), 200
  2764. except Exception as e:
  2765. # 捕获所有异常,确保健康检查不会返回500错误
  2766. logger.error(f"健康检查端点出错: {str(e)}")
  2767. # 返回一个基础的健康状态,至少显示服务在运行
  2768. return jsonify({
  2769. 'status': 'error',
  2770. 'timestamp': time.strftime('%Y-%m-%d %H:%M:%S'),
  2771. 'error': str(e),
  2772. 'services': {
  2773. 'api': True, # API服务本身是运行的
  2774. 'serial': None,
  2775. 'mqtt': None,
  2776. 'websocket': None
  2777. }
  2778. }), 200 # 仍然返回200,避免健康检查导致的连锁反应
  2779. @app.route('/api/screen/status', methods=['GET'])
  2780. def get_screen_status():
  2781. """获取 Nextion 屏幕状态。"""
  2782. with screen_lock:
  2783. panels = get_sorted_panels()
  2784. current_index = screen_current_panel_index
  2785. current_panel_id = None
  2786. if panels and 0 <= current_index < len(panels):
  2787. current_panel_id = panels[current_index][0]
  2788. panel_count = len(panels)
  2789. status = screen_display.get_status()
  2790. return jsonify({
  2791. 'success': True,
  2792. 'data': {
  2793. 'connected': status.get('connected', False),
  2794. 'port': status.get('port'),
  2795. 'baudrate': status.get('baudrate'),
  2796. 'current_panel_index': current_index,
  2797. 'current_panel_id': current_panel_id,
  2798. 'panel_count': panel_count
  2799. }
  2800. })
  2801. @app.route('/api/screen/refresh', methods=['POST'])
  2802. def refresh_screen():
  2803. """手动刷新当前 panel 到屏幕。"""
  2804. with screen_lock:
  2805. panels = get_sorted_panels()
  2806. if not panels:
  2807. return jsonify({'success': False, 'message': '没有可用的 panel'}), 400
  2808. if not (0 <= screen_current_panel_index < len(panels)):
  2809. return jsonify({'success': False, 'message': '当前 panel 索引无效'}), 400
  2810. panel_id = panels[screen_current_panel_index][0]
  2811. success, message = refresh_screen_panel(panel_id)
  2812. if success:
  2813. return jsonify({'success': True, 'message': message})
  2814. return jsonify({'success': False, 'message': message}), 400
  2815. @app.route('/api/screen/test_cmd', methods=['POST'])
  2816. def screen_test_cmd():
  2817. """向 Nextion 屏幕发送原始命令(调试用)。"""
  2818. try:
  2819. data = request.json
  2820. cmd = data.get('cmd')
  2821. if not cmd:
  2822. return jsonify({'success': False, 'message': '缺少 cmd 参数'}), 400
  2823. status = screen_display.get_status()
  2824. if not status.get('connected'):
  2825. return jsonify({'success': False, 'message': '屏幕串口未连接'}), 400
  2826. ok = screen_display.send_cmd(cmd, delay_ms=data.get('delay_ms', 50))
  2827. logger.info(f"[Nextion TEST] cmd={cmd} ok={ok}")
  2828. return jsonify({'success': ok, 'cmd': cmd, 'screen_connected': True})
  2829. except Exception as e:
  2830. logger.error(f"屏幕测试命令失败: {e}")
  2831. return jsonify({'success': False, 'message': str(e)}), 500
  2832. # 网络配置相关API
  2833. @app.route('/api/network/config', methods=['GET'])
  2834. def get_network_config():
  2835. """获取当前网络配置"""
  2836. try:
  2837. config = network_manager.get_network_config()
  2838. return jsonify(config)
  2839. except Exception as e:
  2840. logger.error(f'获取网络配置失败: {str(e)}')
  2841. return jsonify({'success': False, 'message': f'获取网络配置失败: {str(e)}'}), 500
  2842. @app.route('/api/network/config', methods=['POST'])
  2843. def update_network_config():
  2844. """更新网络配置"""
  2845. try:
  2846. config_data = request.json
  2847. result = network_manager.update_network_config(config_data)
  2848. if result['success']:
  2849. return jsonify(result)
  2850. else:
  2851. return jsonify(result), 400
  2852. except Exception as e:
  2853. logger.error(f'更新网络配置失败: {str(e)}')
  2854. return jsonify({'success': False, 'message': f'更新网络配置失败: {str(e)}'}), 500
  2855. @app.route('/api/network/status', methods=['GET'])
  2856. def get_network_status():
  2857. """获取网络状态信息"""
  2858. try:
  2859. status = network_manager.get_network_status()
  2860. return jsonify(status)
  2861. except Exception as e:
  2862. logger.error(f'获取网络状态失败: {str(e)}')
  2863. return jsonify({'success': False, 'message': f'获取网络状态失败: {str(e)}'}), 500
  2864. @app.route('/api/network/restart', methods=['POST'])
  2865. def restart_network_service():
  2866. """重启网络服务以应用新配置"""
  2867. try:
  2868. result = network_manager.restart_network_service()
  2869. if result['success']:
  2870. return jsonify(result)
  2871. else:
  2872. return jsonify(result), 500
  2873. except Exception as e:
  2874. logger.error(f'重启网络服务失败: {str(e)}')
  2875. return jsonify({'success': False, 'message': f'重启网络服务失败: {str(e)}'}), 500
  2876. # Modbus RTU API
  2877. @app.route('/api/modbus/antenna_addresses', methods=['GET'])
  2878. def get_antenna_addresses():
  2879. """获取天线地址映射表"""
  2880. return jsonify({
  2881. 'success': True,
  2882. 'antennas': {str(k): f"0x{v:04x}" for k, v in ANTENNA_ADDRESSES.items()}
  2883. })
  2884. @app.route('/api/modbus/read_antenna', methods=['POST'])
  2885. def modbus_read_antenna():
  2886. """读取指定天线的卡号
  2887. 请求参数:
  2888. {
  2889. "device_address": 1, // 设备地址 (1-247)
  2890. "antenna": 1, // 天线编号 (1-24)
  2891. "timeout": 1.0 // 可选,超时时间(秒)
  2892. }
  2893. """
  2894. try:
  2895. data = request.json
  2896. device_address = data.get('device_address', 1)
  2897. antenna = data.get('antenna', 1)
  2898. timeout = data.get('timeout')
  2899. if antenna < 1 or antenna > 24:
  2900. return jsonify({
  2901. 'success': False,
  2902. 'message': f'无效的天线编号: {antenna}, 必须是1-24'
  2903. }), 400
  2904. if not (isinstance(serial_client.get_status(), dict) and serial_client.get_status().get("connected", False)):
  2905. return jsonify({
  2906. 'success': False,
  2907. 'message': '串口未连接'
  2908. }), 400
  2909. result = modbus_client.read_antenna_card(
  2910. device_address=device_address,
  2911. antenna_num=antenna,
  2912. timeout=timeout
  2913. )
  2914. if 'error' in result:
  2915. return jsonify({
  2916. 'success': False,
  2917. 'message': result['error'],
  2918. 'raw_data': result.get('raw_data', '')
  2919. }), 400
  2920. return jsonify({
  2921. 'success': True,
  2922. 'data': result
  2923. })
  2924. except Exception as e:
  2925. logger.error(f'读取天线数据失败: {str(e)}')
  2926. return jsonify({
  2927. 'success': False,
  2928. 'message': str(e)
  2929. }), 500
  2930. @app.route('/api/modbus/read_all_cards', methods=['POST'])
  2931. def modbus_read_all_cards():
  2932. """V4: 一次批量读取全部 24 路天线卡号
  2933. 请求参数:
  2934. {
  2935. "device_address": 2,
  2936. "timeout": 1.0
  2937. }
  2938. 返回:
  2939. {
  2940. "success": true,
  2941. "data": {
  2942. "device_address": 2,
  2943. "cards": [
  2944. {"antenna": 1, "present": false, "card_str": "", "uid": ""},
  2945. {"antenna": 2, "present": true, "card_str": "aabbccddeeff", "uid": "aa:bb:cc:dd:ee:ff"},
  2946. ...
  2947. ]
  2948. }
  2949. }
  2950. """
  2951. try:
  2952. data = request.json or {}
  2953. device_address = data.get('device_address', 1)
  2954. timeout = data.get('timeout')
  2955. if not (isinstance(serial_client.get_status(), dict) and serial_client.get_status().get("connected", False)):
  2956. return jsonify({'success': False, 'message': '串口未连接'}), 400
  2957. result = modbus_client.read_all_antenna_cards(device_address, timeout=timeout)
  2958. if 'error' in result:
  2959. return jsonify({
  2960. 'success': False,
  2961. 'message': result['error'],
  2962. 'raw_data': result.get('raw_data', '')
  2963. }), 400
  2964. return jsonify({'success': True, 'data': result})
  2965. except Exception as e:
  2966. logger.error(f'批量读卡失败: {str(e)}')
  2967. return jsonify({'success': False, 'message': str(e)}), 500
  2968. @app.route('/api/modbus/set_card_mapping', methods=['POST'])
  2969. def modbus_set_card_mapping():
  2970. """V4: 设置24路天线卡号到配线架硬件(0x10 写 0x0003,144字节)
  2971. 请求参数:
  2972. {
  2973. "device_address": 1,
  2974. "card_mappings": {"1": "aabbccddeeff", "2": null, ...}, // 完整模式:24路映射
  2975. // 或单端口模式:
  2976. "port_id": 1,
  2977. "card_uid": "aabbccddeeff", // null/空表示清除该端口卡号
  2978. "panel_id": "PANEL_dtu_001_2", // 单端口模式下用于读取其余端口期望卡号(可选)
  2979. "timeout": 2.0
  2980. }
  2981. 返回:
  2982. {
  2983. "success": true,
  2984. "data": {"device_address": 1, "cards_written": {...}, "raw_data": "..."}
  2985. }
  2986. """
  2987. try:
  2988. data = request.json or {}
  2989. device_address = data.get('device_address', 1)
  2990. timeout = data.get('timeout')
  2991. if not (isinstance(serial_client.get_status(), dict) and serial_client.get_status().get("connected", False)):
  2992. return jsonify({'success': False, 'message': '串口未连接'}), 400
  2993. card_mappings = data.get('card_mappings')
  2994. if card_mappings is None:
  2995. # 单端口模式:构建 24 路数据,避免擦掉其他端口
  2996. # 策略: 先读硬件当前值做基底 → 再用 dtu_config 期望值覆盖 → 最后覆盖目标端口
  2997. # 这样 dtu_config 没记录但硬件已有卡号的端口(例如修复前设置的)不会被误清为 0xFF
  2998. port_id = data.get('port_id')
  2999. card_uid = data.get('card_uid')
  3000. # panel_id 解析:前端传的优先,否则用 device_address 反查 panel_config(LedDebug.vue 默认只传 address)
  3001. panel_id_from_req = _resolve_panel_id(data.get('panel_id'))
  3002. panel_id_from_addr = None
  3003. for pid, cfg in panel_config.items():
  3004. if cfg.get('address') == device_address:
  3005. panel_id_from_addr = pid
  3006. break
  3007. panel_id = panel_id_from_req or panel_id_from_addr
  3008. if port_id is None:
  3009. return jsonify({'success': False, 'message': '缺少 port_id 或 card_mappings'}), 400
  3010. # 第一步:读硬件当前 24 路值,作为默认基底(防止擦掉其他端口硬件里真实存在的卡号)
  3011. card_mappings = {i: None for i in range(1, 25)}
  3012. try:
  3013. hw_read = modbus_client.read_antenna_card(device_address, timeout=2.0)
  3014. if 'error' not in hw_read:
  3015. for card in hw_read.get('cards', []):
  3016. ant = card.get('antenna')
  3017. cs = card.get('card_str') or card.get('uid')
  3018. if ant and cs and str(cs).lower() != 'ffffffffffff':
  3019. cs_clean = str(cs).strip().upper().replace(':', '').replace(' ', '')
  3020. if len(cs_clean) == 12 and all(c in _JUMPER_HEX for c in cs_clean):
  3021. card_mappings[int(ant)] = cs_clean
  3022. logger.debug(f"单端口模式已用硬件读值初始化基底: {sum(1 for v in card_mappings.values() if v)} 路有卡")
  3023. except Exception as e:
  3024. logger.warning(f"单端口模式读取硬件基底失败(将使用 dtu_config 兜底): {e}")
  3025. # 第二步:用 dtu_config 期望值覆盖(平台下发的才是权威,硬件基底只补缺失项)
  3026. # 注意:dtu_config 中已有但硬件读不到的端口(比如刚写入、读延迟),也必须强制合入,防止回擦
  3027. if panel_id:
  3028. dtu_base = _build_card_mappings_for_panel(panel_id)
  3029. for ant, v in dtu_base.items():
  3030. if v is not None:
  3031. card_mappings[ant] = v
  3032. logger.debug(f"单端口模式 panel={panel_id} dtu_config 期望合入: {sum(1 for v in dtu_base.values() if v)} 路")
  3033. # 第三步:最后覆盖目标端口(用户要设置/清除的值优先级最高)
  3034. if card_uid:
  3035. uid = str(card_uid).strip().upper().replace(':', '').replace(' ', '')
  3036. if len(uid) != 12 or not all(c in _JUMPER_HEX for c in uid):
  3037. return jsonify({'success': False, 'message': f'卡号必须是12位十六进制: {card_uid}'}), 400
  3038. else:
  3039. uid = None
  3040. try:
  3041. card_mappings[int(port_id)] = uid
  3042. except (ValueError, TypeError):
  3043. return jsonify({'success': False, 'message': f'port_id 非法: {port_id}'}), 400
  3044. # 转换 key 为 int
  3045. normalized = {}
  3046. for k, v in card_mappings.items():
  3047. try:
  3048. port = int(k)
  3049. except (ValueError, TypeError):
  3050. return jsonify({'success': False, 'message': f'端口编号非法: {k}'}), 400
  3051. if not (1 <= port <= 24):
  3052. return jsonify({'success': False, 'message': f'端口编号必须在1-24: {port}'}), 400
  3053. normalized[port] = v
  3054. result = modbus_client.set_card_mapping(device_address, normalized, timeout=timeout)
  3055. if 'error' in result:
  3056. return jsonify({
  3057. 'success': False,
  3058. 'message': result['error'],
  3059. 'raw_data': result.get('raw_data', '')
  3060. }), 400
  3061. # 硬件写入成功:同步更新本地 port_state 期望值 + 持久化 dtu_config
  3062. # 1. 先通过 device_address 反查 panel_id
  3063. panel_id_by_addr = None
  3064. for pid, cfg in panel_config.items():
  3065. if cfg.get('address') == device_address:
  3066. panel_id_by_addr = pid
  3067. break
  3068. # 单端口模式下若传入了 panel_id,优先用传入值
  3069. panel_target = _resolve_panel_id(data.get('panel_id')) or panel_id_by_addr
  3070. warnings = []
  3071. cards_written = result.get('cards_written', {})
  3072. if panel_target:
  3073. port_state.setdefault(panel_target, {})
  3074. for port_id, uid in cards_written.items():
  3075. port_id = int(port_id)
  3076. port_state[panel_target].setdefault(port_id, {
  3077. 'last_uid': None, 'expected_uid': None, 'alarm_count': 0
  3078. })
  3079. # 规范化为 12 位大写 hex 或 None
  3080. if uid:
  3081. clean_uid = str(uid).strip().upper().replace(':', '').replace(' ', '')
  3082. if len(clean_uid) == 12 and all(c in _JUMPER_HEX for c in clean_uid):
  3083. expected_uid = clean_uid
  3084. else:
  3085. expected_uid = None
  3086. else:
  3087. expected_uid = None
  3088. port_state[panel_target][port_id]['expected_uid'] = expected_uid
  3089. # 持久化到 dtu_config.json(内部 12 位)
  3090. dtu_config.setdefault('port_mappings', {}).setdefault(panel_target, {})
  3091. dtu_config['port_mappings'][panel_target][str(port_id)] = expected_uid
  3092. save_dtu_config()
  3093. logger.info(f"卡号设置页 面板{panel_target} 已同步本地期望值: {len(cards_written)} 路")
  3094. else:
  3095. warnings.append(f"未找到 device_address={device_address} 对应的面板,本地期望值未更新(仅硬件已生效)")
  3096. return jsonify({'success': True, 'data': result, 'warnings': warnings})
  3097. except Exception as e:
  3098. logger.error(f'设置卡号失败: {str(e)}')
  3099. return jsonify({'success': False, 'message': str(e)}), 500
  3100. @app.route('/api/modbus/read_registers', methods=['POST'])
  3101. def modbus_read_registers():
  3102. """读取保持寄存器
  3103. 请求参数:
  3104. {
  3105. "device_address": 1,
  3106. "start_address": 2, // 起始地址 (十六进制如0x0002或十进制如2)
  3107. "quantity": 4, // 寄存器数量
  3108. "timeout": 1.0
  3109. }
  3110. """
  3111. try:
  3112. data = request.json
  3113. device_address = data.get('device_address', 1)
  3114. start_address = data.get('start_address', 0)
  3115. quantity = data.get('quantity', 1)
  3116. timeout = data.get('timeout')
  3117. # 支持十六进制字符串
  3118. if isinstance(start_address, str):
  3119. start_address = int(start_address, 16)
  3120. if isinstance(device_address, str):
  3121. device_address = int(device_address, 16)
  3122. if not (isinstance(serial_client.get_status(), dict) and serial_client.get_status().get("connected", False)):
  3123. return jsonify({
  3124. 'success': False,
  3125. 'message': '串口未连接'
  3126. }), 400
  3127. result = modbus_client.read_holding_registers(
  3128. device_address=device_address,
  3129. start_address=start_address,
  3130. quantity=quantity,
  3131. timeout=timeout
  3132. )
  3133. if 'error' in result:
  3134. return jsonify({
  3135. 'success': False,
  3136. 'message': result['error'],
  3137. 'raw_data': result.get('raw_data', '')
  3138. }), 400
  3139. return jsonify({
  3140. 'success': True,
  3141. 'data': result
  3142. })
  3143. except Exception as e:
  3144. logger.error(f'读取寄存器失败: {str(e)}')
  3145. return jsonify({
  3146. 'success': False,
  3147. 'message': str(e)
  3148. }), 500
  3149. @app.route('/api/modbus/write_register', methods=['POST'])
  3150. def modbus_write_register():
  3151. """写单个寄存器
  3152. 请求参数:
  3153. {
  3154. "device_address": 1,
  3155. "register_address": 1, // 寄存器地址
  3156. "value": 256, // 写入的值
  3157. "timeout": 1.0
  3158. }
  3159. """
  3160. try:
  3161. data = request.json
  3162. device_address = data.get('device_address', 1)
  3163. register_address = data.get('register_address', 1)
  3164. value = data.get('value', 0)
  3165. timeout = data.get('timeout')
  3166. if not (isinstance(serial_client.get_status(), dict) and serial_client.get_status().get("connected", False)):
  3167. return jsonify({
  3168. 'success': False,
  3169. 'message':'串口未连接'
  3170. }), 400
  3171. result = modbus_client.write_single_register(
  3172. device_address=device_address,
  3173. register_address=register_address,
  3174. value=value,
  3175. timeout=timeout
  3176. )
  3177. if 'error' in result:
  3178. return jsonify({
  3179. 'success': False,
  3180. 'message': result['error'],
  3181. 'raw_data': result.get('raw_data', '')
  3182. }), 400
  3183. return jsonify({
  3184. 'success': True,
  3185. 'data': result
  3186. })
  3187. except Exception as e:
  3188. logger.error(f'写寄存器失败: {str(e)}')
  3189. return jsonify({
  3190. 'success': False,
  3191. 'message': str(e)
  3192. }), 500
  3193. @app.route('/api/modbus/set_rgb_led', methods=['POST'])
  3194. def modbus_set_rgb_led():
  3195. """设置RGB灯状态
  3196. 请求参数:
  3197. {
  3198. "device_address": 1,
  3199. "led_number": 1, // 灯编号 (1-24)
  3200. "color": 1, // 颜色: 0=灭, 1=红灯, 2=绿灯, 3=蓝灯
  3201. "timeout": 1.0
  3202. }
  3203. """
  3204. try:
  3205. data = request.json
  3206. device_address = data.get('device_address', 1)
  3207. led_number = data.get('led_number', 1)
  3208. color = data.get('color', 0)
  3209. timeout = data.get('timeout')
  3210. if not (isinstance(serial_client.get_status(), dict) and serial_client.get_status().get("connected", False)):
  3211. return jsonify({
  3212. 'success': False,
  3213. 'message': '串口未连接'
  3214. }), 400
  3215. result = modbus_client.set_rgb_led(
  3216. device_address=device_address,
  3217. led_number=led_number,
  3218. color=color,
  3219. timeout=timeout
  3220. )
  3221. if 'error' in result:
  3222. return jsonify({
  3223. 'success': False,
  3224. 'message': result['error'],
  3225. 'raw_data': result.get('raw_data', '')
  3226. }), 400
  3227. # 按设备地址更新 LED 状态追踪
  3228. if device_address not in led_states:
  3229. led_states[device_address] = {}
  3230. led_states[device_address][led_number] = color
  3231. _refresh_current_screen_panel()
  3232. return jsonify({
  3233. 'success': True,
  3234. 'data': result
  3235. })
  3236. except Exception as e:
  3237. logger.error(f'设置RGB灯失败: {str(e)}')
  3238. return jsonify({
  3239. 'success': False,
  3240. 'message': str(e)
  3241. }), 500
  3242. # ========== 测试注入 API(模拟串口假数据,用于告警/同步映射端到端测试) ==========
  3243. @app.route('/api/test/inject_cards', methods=['POST'])
  3244. def test_inject_cards():
  3245. """注入假卡号,绕过真实串口。
  3246. 请求 body:
  3247. {
  3248. "cards": {
  3249. "2": {"1": "aabbccddeeff", "5": null, "12": "112233445566"}
  3250. }
  3251. }
  3252. key 为设备地址(Modbus 从机地址),value 为 {port_id: hex12|null}。
  3253. 调用后轮询将读到构造的卡号,触发 event/alarm 流程。
  3254. """
  3255. try:
  3256. data = request.json or {}
  3257. raw = data.get('cards')
  3258. if raw is None:
  3259. modbus_client.test_cards = None
  3260. return jsonify({'success': True, 'message': '注入已清空,恢复真实串口'})
  3261. parsed = {}
  3262. for addr_str, port_map in raw.items():
  3263. addr = int(addr_str)
  3264. pm = {}
  3265. for pid_str, hex_str in (port_map or {}).items():
  3266. pid = int(pid_str)
  3267. if hex_str in (None, '', 'null'):
  3268. pm[pid] = None
  3269. else:
  3270. s = hex_str.lower()
  3271. if len(s) != 12:
  3272. return jsonify({'success': False, 'message': f'hex 必须 12 字符: addr={addr} port={pid}'}), 400
  3273. bytes.fromhex(s) # 校验
  3274. pm[pid] = s
  3275. parsed[addr] = pm
  3276. modbus_client.test_cards = parsed
  3277. return jsonify({'success': True, 'test_cards': {
  3278. str(a): {str(p): v for p, v in m.items()} for a, m in parsed.items()
  3279. }})
  3280. except Exception as e:
  3281. return jsonify({'success': False, 'message': str(e)}), 400
  3282. @app.route('/api/test/inject_cards', methods=['GET'])
  3283. def test_inject_cards_get():
  3284. """查看当前注入状态"""
  3285. tc = modbus_client.test_cards
  3286. if tc is None:
  3287. return jsonify({'success': True, 'injected': False})
  3288. return jsonify({
  3289. 'success': True, 'injected': True,
  3290. 'cards': {str(a): {str(p): v for p, v in m.items()} for a, m in tc.items()}
  3291. })
  3292. @app.route('/api/modbus/scan', methods=['POST'])
  3293. def modbus_scan_devices():
  3294. """扫描在线设备
  3295. 请求参数:
  3296. {
  3297. "max_address": 247, // 最大设备地址
  3298. "timeout": 0.2 // 单个设备超时时间
  3299. }
  3300. """
  3301. try:
  3302. data = request.json or {}
  3303. max_address = data.get('max_address', 247)
  3304. timeout = data.get('timeout', 0.2)
  3305. if not (isinstance(serial_client.get_status(), dict) and serial_client.get_status().get("connected", False)):
  3306. return jsonify({
  3307. 'success': False,
  3308. 'message': '串口未连接'
  3309. }), 400
  3310. # 异步扫描可能更好,但这里先同步实现
  3311. result = modbus_client.scan_devices(max_address)
  3312. return jsonify({
  3313. 'success': True,
  3314. 'devices': result
  3315. })
  3316. except Exception as e:
  3317. logger.error(f'扫描设备失败: {str(e)}')
  3318. return jsonify({
  3319. 'success': False,
  3320. 'message': str(e)
  3321. }), 500
  3322. # ========== 地址配置协议 API ==========
  3323. @app.route('/api/modbus/broadcast_query', methods=['POST'])
  3324. def modbus_broadcast_query():
  3325. """发送广播查询指令"""
  3326. try:
  3327. data = request.json or {}
  3328. timeout = data.get('timeout')
  3329. if not (isinstance(serial_client.get_status(), dict) and serial_client.get_status().get("connected", False)):
  3330. return jsonify({'success': False, 'message': '串口未连接'}), 400
  3331. responses = address_config.broadcast_query(timeout)
  3332. assign_results = []
  3333. if responses:
  3334. assign_results = address_config.process_responses(responses)
  3335. save_device_config()
  3336. sync_panel_config_from_results(assign_results)
  3337. return jsonify({
  3338. 'success': True,
  3339. 'responses': responses,
  3340. 'assign_results': assign_results,
  3341. 'count': len(responses)
  3342. })
  3343. except Exception as e:
  3344. logger.error(f'广播查询失败: {str(e)}')
  3345. return jsonify({'success': False, 'message': str(e)}), 500
  3346. @app.route('/api/modbus/auto_configure', methods=['POST'])
  3347. def modbus_auto_configure():
  3348. """自动配置设备地址"""
  3349. try:
  3350. data = request.json or {}
  3351. timeout = data.get('timeout')
  3352. if not (isinstance(serial_client.get_status(), dict) and serial_client.get_status().get("connected", False)):
  3353. return jsonify({'success': False, 'message': '串口未连接'}), 400
  3354. result = address_config.auto_configure(timeout)
  3355. if result.get('success') and (result.get('discovered', 0) > 0 or result.get('assigned', 0) > 0):
  3356. save_device_config()
  3357. sync_panel_config_from_results(result.get('results', []))
  3358. return jsonify(result)
  3359. except Exception as e:
  3360. logger.error(f'自动配置失败: {str(e)}')
  3361. return jsonify({'success': False, 'message': str(e)}), 500
  3362. @app.route('/api/modbus/stored_devices', methods=['GET'])
  3363. def get_stored_devices():
  3364. """获取已注册线架列表(地址↔UID 绑定 + 在线/缺失状态)"""
  3365. devices = address_config.get_stored_devices()
  3366. now = time.time()
  3367. online_threshold = 60 # 60 秒内被轮询到视为在线
  3368. panels = []
  3369. for uid, addr in devices.items():
  3370. pid = _panel_id_for_address(addr)
  3371. ps = port_state.get(pid, {})
  3372. last_seen = device_last_seen.get(pid) or device_last_seen.get(uid)
  3373. recent = bool(last_seen) and (now - last_seen) < online_threshold
  3374. missing = panel_fail_count.get(pid, 0) > 0
  3375. status = 'online' if (recent and not missing) else ('missing' if missing else 'offline')
  3376. panels.append({
  3377. 'uid': uid,
  3378. 'address': addr,
  3379. 'online': status == 'online',
  3380. 'panel_id': pid,
  3381. 'status': status,
  3382. 'last_seen': last_seen,
  3383. 'port_count': len(ps),
  3384. 'connected_count': sum(1 for p in ps.values() if p.get('last_uid')),
  3385. 'alarm_count': sum(1 for p in ps.values() if p.get('alarm_count', 0) > 0),
  3386. })
  3387. return jsonify({
  3388. 'success': True,
  3389. 'devices': devices,
  3390. 'panels': panels,
  3391. 'address_range': {'start': _address_range()[0], 'end': _address_range()[1]},
  3392. 'scan': _panel_scan_state
  3393. })
  3394. @app.route('/api/modbus/stored_devices', methods=['POST'])
  3395. def add_stored_device():
  3396. """添加已存储的设备"""
  3397. try:
  3398. data = request.json
  3399. uid = data.get('uid', '').lower()
  3400. address = data.get('address', 1)
  3401. if len(uid) != 24:
  3402. return jsonify({'success': False, 'message': 'UID长度必须是24个十六进制字符'}), 400
  3403. address_config.add_stored_device(uid, address)
  3404. save_device_config()
  3405. return jsonify({'success': True, 'message': f'已添加设备: UID={uid}, 地址={address}'})
  3406. except Exception as e:
  3407. return jsonify({'success': False, 'message': str(e)}), 500
  3408. @app.route('/api/panel/unbind', methods=['POST'])
  3409. def panel_unbind():
  3410. """解除线架绑定(仅解除;线架仍在总线上时,下轮地址段扫描会重新识别绑定)"""
  3411. try:
  3412. data = request.json or {}
  3413. addr = data.get('address')
  3414. panel_id = data.get('panel_id')
  3415. if addr is None and panel_id:
  3416. cfg = panel_config.get(panel_id)
  3417. if cfg:
  3418. addr = cfg.get('address')
  3419. elif isinstance(panel_id, str) and panel_id.rsplit('_', 1)[-1].isdigit():
  3420. addr = int(panel_id.rsplit('_', 1)[-1])
  3421. if not isinstance(addr, int):
  3422. return jsonify({'success': False, 'message': '缺少有效的 address 或 panel_id'}), 400
  3423. uid = _binding_uid_for_address(addr)
  3424. if uid:
  3425. address_config.stored_devices.pop(uid, None)
  3426. save_device_config()
  3427. pid = panel_id if (panel_id and panel_id in panel_config) else _panel_id_for_address(addr)
  3428. panel_config.pop(pid, None)
  3429. port_state.pop(pid, None)
  3430. panel_fail_count.pop(pid, None)
  3431. device_last_seen.pop(pid, None)
  3432. if uid:
  3433. device_last_seen.pop(uid, None)
  3434. logger.info(f"已解除线架绑定: 地址={addr} UID={uid or '无'}")
  3435. try:
  3436. dtu_publish_panel_status(pid, addr, None, online=False)
  3437. except Exception:
  3438. pass
  3439. return jsonify({
  3440. 'success': True,
  3441. 'message': f"已解除地址 {addr} 的绑定" + (f"(UID={uid})" if uid else "(无绑定记录)"),
  3442. 'address': addr,
  3443. 'uid': uid
  3444. })
  3445. except Exception as e:
  3446. logger.error(f"解除线架绑定失败: {e}")
  3447. return jsonify({'success': False, 'message': str(e)}), 500
  3448. def _run_panel_scan():
  3449. """后台扫描地址段内全部未绑定地址(手动触发)"""
  3450. global _panel_scan_state
  3451. _panel_scan_state = {'running': True, 'found': [], 'started_at': time.time(),
  3452. 'finished_at': None, 'error': None}
  3453. try:
  3454. with address_config.config_lock:
  3455. start, end = _address_range()
  3456. bound_addrs = set(address_config.get_stored_devices().values())
  3457. for addr in range(start, end + 1):
  3458. if addr in bound_addrs:
  3459. continue
  3460. result = modbus_client.read_all_antenna_cards(addr, uid=None, timeout=PANEL_PROBE_TIMEOUT)
  3461. if 'error' not in result:
  3462. new_uid = (result.get('panel_uid') or '').lower()
  3463. if new_uid and _register_new_panel(addr, new_uid, result):
  3464. _panel_scan_state['found'].append({'address': addr, 'uid': new_uid})
  3465. bound_addrs.add(addr)
  3466. time.sleep(0.05)
  3467. except Exception as e:
  3468. logger.error(f"线架扫描失败: {e}")
  3469. _panel_scan_state['error'] = str(e)
  3470. finally:
  3471. _panel_scan_state['running'] = False
  3472. _panel_scan_state['finished_at'] = time.time()
  3473. logger.info(f"线架扫描完成: 发现 {len(_panel_scan_state.get('found', []))} 台")
  3474. @app.route('/api/panel/scan', methods=['POST'])
  3475. def panel_scan():
  3476. """手动扫描地址段内未绑定地址,发现并注册新线架"""
  3477. if _panel_scan_state.get('running'):
  3478. return jsonify({'success': False, 'message': '扫描正在进行中,请稍候'}), 409
  3479. threading.Thread(target=_run_panel_scan, daemon=True).start()
  3480. return jsonify({'success': True, 'message': '已开始扫描地址段内未绑定地址'})
  3481. @app.route('/api/panel/scan_status', methods=['GET'])
  3482. def panel_scan_status():
  3483. """查询扫描进度/结果"""
  3484. return jsonify({'success': True, 'scan': _panel_scan_state})
  3485. """从文件加载设备配置"""
  3486. try:
  3487. data = request.json
  3488. filepath = data.get('filepath', '/tmp/modbus_devices.json')
  3489. success = address_config.load_config(filepath)
  3490. return jsonify({'success': success, 'devices': address_config.get_stored_devices()})
  3491. except Exception as e:
  3492. return jsonify({'success': False, 'message': str(e)}), 500
  3493. @app.route('/api/modbus/save_config', methods=['POST'])
  3494. def modbus_save_config():
  3495. """保存设备配置到文件"""
  3496. try:
  3497. data = request.json
  3498. filepath = data.get('filepath', '/tmp/modbus_devices.json')
  3499. success = address_config.save_config(filepath)
  3500. return jsonify({'success': success, 'message': f'配置已保存到: {filepath}' if success else '保存失败'})
  3501. except Exception as e:
  3502. return jsonify({'success': False, 'message': str(e)}), 500
  3503. @app.route('/api/modbus/confirm_devices', methods=['POST'])
  3504. def confirm_devices():
  3505. """手动触发对所有已存储设备的确认"""
  3506. try:
  3507. devices = address_config.get_stored_devices()
  3508. if not devices:
  3509. return jsonify({'success': False, 'message': '没有已存储的设备'}), 400
  3510. if not (isinstance(serial_client.get_status(), dict) and serial_client.get_status().get("connected", False)):
  3511. return jsonify({'success': False, 'message': '串口未连接'}), 400
  3512. results = []
  3513. for uid_hex, addr in devices.items():
  3514. try:
  3515. uid_bytes = bytes.fromhex(uid_hex)
  3516. cmd = build_confirm_address(addr, uid_bytes)
  3517. success, msg = serial_client.send_raw(cmd)
  3518. logger.info(f"确认设备 地址={addr}: {'成功' if success else '失败 ' + msg}")
  3519. results.append({'address': addr, 'uid': uid_hex, 'success': success, 'message': msg})
  3520. except Exception as e:
  3521. results.append({'address': addr, 'uid': uid_hex, 'success': False, 'message': str(e)})
  3522. time.sleep(0.1)
  3523. return jsonify({'success': True, 'results': results, 'count': len(results)})
  3524. except Exception as e:
  3525. return jsonify({'success': False, 'message': str(e)}), 500
  3526. # 端口状态追踪 - 用于存储每个端口的事件状态
  3527. port_event_history = [] # 存储最近的端口事件
  3528. @app.route('/api/port/status', methods=['GET'])
  3529. def get_port_status():
  3530. """获取所有端口状态"""
  3531. return jsonify({'success': True, 'port_state': port_state})
  3532. @app.route('/api/port/events', methods=['GET'])
  3533. def get_port_events():
  3534. """获取端口事件历史"""
  3535. limit = request.args.get('limit', 100, type=int)
  3536. return jsonify({
  3537. 'success': True,
  3538. 'events': port_event_history[-limit:]
  3539. })
  3540. @app.route('/api/port/clear_events', methods=['POST'])
  3541. def clear_port_events():
  3542. """清除端口事件历史"""
  3543. global port_event_history
  3544. port_event_history = []
  3545. return jsonify({'success': True, 'message': '事件历史已清除'})
  3546. # 设备最后响应时间跟踪
  3547. device_last_seen = {}
  3548. @app.route('/api/panel/status', methods=['GET'])
  3549. def get_panel_status():
  3550. """获取所有面板状态,包含每个端口的连接状态"""
  3551. now = time.time()
  3552. PANEL_OFFLINE_TIMEOUT = 60
  3553. panel_status = {}
  3554. for panel_id, ports in port_state.items():
  3555. port_count = len(ports)
  3556. alarm_count = sum(1 for p in ports.values() if p.get('alarm_count', 0) > 0)
  3557. connected_count = sum(1 for p in ports.values() if p.get('last_uid'))
  3558. panel_ports = {}
  3559. for port_id, p in ports.items():
  3560. last_uid = p.get('last_uid')
  3561. expected_uid = p.get('expected_uid')
  3562. last_polled_at = p.get('last_polled_at')
  3563. if not last_polled_at:
  3564. status = 'UNKNOWN'
  3565. elif not last_uid:
  3566. # 有期望但读不到卡:告警未插;无期望:正常空闲
  3567. status = 'ILLEGAL_DISCONNECT' if expected_uid else 'DISCONNECTED'
  3568. elif expected_uid and last_uid != expected_uid:
  3569. status = 'ILLEGAL_CONNECT'
  3570. else:
  3571. status = 'CONNECTED'
  3572. panel_ports[port_id] = {
  3573. 'status': status,
  3574. 'jumper_uid': last_uid or None,
  3575. 'expected_uid': expected_uid or None,
  3576. 'alarm_count': p.get('alarm_count', 0)
  3577. }
  3578. panel_status[panel_id] = {
  3579. 'panel_id': panel_id,
  3580. 'port_count': port_count,
  3581. 'connected_count': connected_count,
  3582. 'alarm_count': alarm_count,
  3583. 'status': 'online' if connected_count > 0 else 'offline',
  3584. 'ports': panel_ports
  3585. }
  3586. for panel_id, cfg in panel_config.items():
  3587. last_seen = device_last_seen.get(panel_id, 0)
  3588. is_online = (now - last_seen) < PANEL_OFFLINE_TIMEOUT
  3589. if panel_id in panel_status:
  3590. panel_status[panel_id].update({
  3591. 'address': cfg.get('address'),
  3592. 'position': cfg.get('position'),
  3593. 'panel_uid': cfg.get('panel_uid'),
  3594. 'status': 'online' if is_online else 'offline'
  3595. })
  3596. else:
  3597. panel_status[panel_id] = {
  3598. 'panel_id': panel_id,
  3599. 'address': cfg.get('address'),
  3600. 'position': cfg.get('position'),
  3601. 'panel_uid': cfg.get('panel_uid'),
  3602. 'port_count': 0,
  3603. 'connected_count': 0,
  3604. 'alarm_count': 0,
  3605. 'status': 'online' if is_online else 'offline',
  3606. 'ports': {}
  3607. }
  3608. return jsonify({'success': True, 'panels': panel_status})
  3609. # LED 状态追踪 (按设备地址跟踪每个灯的最后设置状态)
  3610. # 结构: {device_address: {port_id: color}}, color: 0=off, 1=red, 2=green, 3=blue
  3611. led_states = {}
  3612. # Nextion 串口屏网口图片与 LED 颜色的映射关系
  3613. # 屏幕图片定义: 0=灰(灭), 1=蓝, 2=红, 3=绿
  3614. # 后端 LED 颜色定义: 0=灭, 1=红, 2=绿, 3=蓝
  3615. LED_COLOR_TO_PIC = {0: 0, 1: 2, 2: 3, 3: 1}
  3616. @app.route('/api/modbus/led_status', methods=['GET'])
  3617. def get_led_status():
  3618. """获取 LED 状态,支持按设备地址过滤"""
  3619. device_address = request.args.get('device_address', type=int)
  3620. if device_address is not None:
  3621. return jsonify({
  3622. 'success': True,
  3623. 'device_address': device_address,
  3624. 'leds': led_states.get(device_address, {})
  3625. })
  3626. return jsonify({'success': True, 'leds': led_states})
  3627. @app.route('/api/modbus/set_all_leds', methods=['POST'])
  3628. def set_all_leds():
  3629. """批量设置 LED
  3630. 请求: {"device_address": 1, "color": 1}
  3631. 设置所有 24 个 LED 到指定颜色
  3632. """
  3633. try:
  3634. data = request.json
  3635. device_address = data.get('device_address', 1)
  3636. color = data.get('color', 0)
  3637. timeout = data.get('timeout')
  3638. if not (isinstance(serial_client.get_status(), dict) and serial_client.get_status().get("connected", False)):
  3639. return jsonify({'success': False, 'message': '串口未连接'}), 400
  3640. results = []
  3641. if device_address not in led_states:
  3642. led_states[device_address] = {}
  3643. for led_num in range(1, 25):
  3644. result = modbus_client.set_rgb_led(
  3645. device_address=device_address,
  3646. led_number=led_num,
  3647. color=color,
  3648. timeout=timeout
  3649. )
  3650. if 'error' in result:
  3651. results.append({'led': led_num, 'success': False, 'error': result['error']})
  3652. else:
  3653. results.append({'led': led_num, 'success': True})
  3654. led_states[device_address][led_num] = color
  3655. _refresh_current_screen_panel()
  3656. return jsonify({'success': True, 'results': results})
  3657. except Exception as e:
  3658. logger.error(f'批量设置LED失败: {str(e)}')
  3659. return jsonify({'success': False, 'message': str(e)}), 500
  3660. # 在 set_rgb_led 中更新 LED 状态追踪
  3661. # 不再需要静态文件目录,前端由nginx提供服务
  3662. # ========== DTU MQTT协议配置API ==========
  3663. @app.route('/api/dtu/config', methods=['GET'])
  3664. def get_dtu_config():
  3665. """获取DTU配置"""
  3666. return jsonify({
  3667. 'success': True,
  3668. 'data': dtu_config
  3669. })
  3670. @app.route('/api/dtu/config', methods=['POST'])
  3671. def update_dtu_config():
  3672. """更新DTU配置"""
  3673. try:
  3674. data = request.json
  3675. old_config = dtu_config.copy()
  3676. # 更新配置项
  3677. if 'topic_prefix' in data:
  3678. dtu_config['topic_prefix'] = data['topic_prefix']
  3679. if 'customer_id' in data:
  3680. dtu_config['customer_id'] = data['customer_id']
  3681. if 'dtu_id' in data:
  3682. dtu_config['dtu_id'] = data['dtu_id']
  3683. if 'firmware_version' in data:
  3684. dtu_config['firmware_version'] = data['firmware_version']
  3685. if 'hardware_version' in data:
  3686. dtu_config['hardware_version'] = data['hardware_version']
  3687. if 'heartbeat_interval' in data:
  3688. dtu_config['heartbeat_interval'] = data['heartbeat_interval']
  3689. if 'enabled' in data:
  3690. dtu_config['enabled'] = data['enabled']
  3691. if 'address_range' in data:
  3692. rng = data['address_range']
  3693. err = None
  3694. if not isinstance(rng, dict):
  3695. err = 'address_range 必须是对象'
  3696. else:
  3697. start, end = rng.get('start'), rng.get('end')
  3698. if not isinstance(start, int) or not isinstance(end, int):
  3699. err = 'address_range.start/end 必须是整数'
  3700. elif not (1 <= start <= end <= 247):
  3701. err = '地址段需满足 1 <= 起始 <= 结束 <= 247'
  3702. if err:
  3703. return jsonify({'success': False, 'message': err}), 400
  3704. dtu_config['address_range'] = {'start': int(rng['start']), 'end': int(rng['end'])}
  3705. # 如果MQTT已连接且主题配置发生变化,重新订阅
  3706. if mqtt_status and dtu_config.get('enabled'):
  3707. old_control_topic = build_dtu_topic(
  3708. old_config.get('customer_id', DEFAULT_CUSTOMER_ID),
  3709. 'dtu',
  3710. old_config.get('dtu_id', DEFAULT_DTU_ID),
  3711. 'control'
  3712. )
  3713. new_control_topic = build_dtu_topic(
  3714. dtu_config['customer_id'],
  3715. 'dtu',
  3716. dtu_config['dtu_id'],
  3717. 'control'
  3718. )
  3719. if old_control_topic != new_control_topic:
  3720. # 取消旧订阅,订阅新主题
  3721. mqtt_client.unsubscribe(old_control_topic)
  3722. mqtt_client.subscribe(new_control_topic)
  3723. logger.info(f"控制主题已更新: {old_control_topic} -> {new_control_topic}")
  3724. # 重新发送注册消息
  3725. dtu_register()
  3726. # 持久化 DTU 配置(此前仅改内存,重启会丢)
  3727. save_dtu_config()
  3728. return jsonify({
  3729. 'success': True,
  3730. 'message': 'DTU配置已更新',
  3731. 'data': dtu_config
  3732. })
  3733. except Exception as e:
  3734. logger.error(f"更新DTU配置失败: {str(e)}")
  3735. return jsonify({'success': False, 'message': str(e)}), 500
  3736. @app.route('/api/dtu/register', methods=['POST'])
  3737. def manual_dtu_register():
  3738. """手动触发DTU注册"""
  3739. try:
  3740. if not mqtt_status:
  3741. return jsonify({'success': False, 'message': 'MQTT未连接'}), 400
  3742. success = dtu_register()
  3743. if success:
  3744. return jsonify({'success': True, 'message': '注册消息已发送'})
  3745. else:
  3746. return jsonify({'success': False, 'message': '注册消息发送失败'}), 500
  3747. except Exception as e:
  3748. logger.error(f"手动触发DTU注册失败: {str(e)}")
  3749. return jsonify({'success': False, 'message': str(e)}), 500
  3750. _cpu_last_times = None
  3751. def _get_cpu_usage():
  3752. """CPU 使用率 (psutil 优先,无 psutil 时回退读 /proc/stat 计算瞬时差值)"""
  3753. global _cpu_last_times
  3754. try:
  3755. import psutil
  3756. return psutil.cpu_percent(interval=None)
  3757. except Exception:
  3758. pass
  3759. try:
  3760. with open('/proc/stat') as f:
  3761. fields = f.readline().split()
  3762. if len(fields) < 5 or not fields[0].startswith('cpu'):
  3763. return None
  3764. vals = [int(x) for x in fields[1:]]
  3765. idle = vals[3] + (vals[4] if len(vals) > 4 else 0)
  3766. total = sum(vals)
  3767. if _cpu_last_times:
  3768. dtotal = total - _cpu_last_times[0]
  3769. didle = idle - _cpu_last_times[1]
  3770. if dtotal > 0:
  3771. pct = round((1.0 - didle / dtotal) * 100, 1)
  3772. _cpu_last_times = (total, idle)
  3773. return pct
  3774. _cpu_last_times = (total, idle)
  3775. return None
  3776. except Exception:
  3777. return None
  3778. def _get_memory_usage():
  3779. """内存使用率 (psutil 优先,无 psutil 时回退读 /proc/meminfo)"""
  3780. try:
  3781. import psutil
  3782. return psutil.virtual_memory().percent
  3783. except Exception:
  3784. pass
  3785. try:
  3786. meminfo = {}
  3787. with open('/proc/meminfo') as f:
  3788. for line in f:
  3789. parts = line.split(':')
  3790. if len(parts) == 2:
  3791. try:
  3792. meminfo[parts[0].strip()] = int(parts[1].split()[0])
  3793. except (ValueError, IndexError):
  3794. continue
  3795. total = meminfo.get('MemTotal', 0)
  3796. available = meminfo.get('MemAvailable')
  3797. if available is None:
  3798. available = meminfo.get('MemFree', 0)
  3799. if total > 0:
  3800. return round((1.0 - available / total) * 100, 1)
  3801. return None
  3802. except Exception:
  3803. return None
  3804. @app.route('/api/dtu/status', methods=['GET'])
  3805. def get_dtu_status():
  3806. """获取DTU状态"""
  3807. try:
  3808. # 获取串口状态
  3809. serial_st = serial_client.get_status()
  3810. # 获取面板状态
  3811. devices = address_config.get_stored_devices()
  3812. # 主板温度: 优先外部上报,否则本地读取
  3813. dtu_temperature = env_sensor_data.get('dtu_temperature')
  3814. if dtu_temperature is None:
  3815. try:
  3816. import psutil
  3817. temps = psutil.sensors_temperatures(fahrenheit=False) or {}
  3818. for chip_name, entries in temps.items():
  3819. for entry in entries:
  3820. if entry.current and entry.current > 0:
  3821. dtu_temperature = round(entry.current, 1)
  3822. break
  3823. if dtu_temperature is not None:
  3824. break
  3825. except Exception:
  3826. pass
  3827. if dtu_temperature is None:
  3828. import glob as _glob
  3829. for tz in sorted(_glob.glob('/sys/class/thermal/thermal_zone*/temp')):
  3830. try:
  3831. with open(tz) as f:
  3832. v = int(f.read().strip())
  3833. if v > 0:
  3834. dtu_temperature = round(v / 1000.0, 1)
  3835. break
  3836. except Exception:
  3837. continue
  3838. status = {
  3839. 'dtu_id': dtu_config.get('dtu_id'),
  3840. 'mqtt_connected': mqtt_status,
  3841. 'serial_connected': isinstance(serial_st, dict) and serial_st.get('connected', False),
  3842. 'dtu_enabled': dtu_config.get('enabled', True),
  3843. 'topic_prefix': dtu_config.get('topic_prefix'),
  3844. 'customer_id': dtu_config.get('customer_id'),
  3845. 'panel_count': len(devices),
  3846. 'cpu_usage': _get_cpu_usage(),
  3847. 'memory_usage': _get_memory_usage(),
  3848. 'temperature': dtu_temperature,
  3849. 'dtu_temperature': dtu_temperature,
  3850. 'firmware_version': dtu_config.get('firmware_version', 'v1.0.0'),
  3851. 'topics': {
  3852. 'register': build_dtu_topic(dtu_config['customer_id'], 'dtu', dtu_config['dtu_id'], 'register'),
  3853. 'status': build_dtu_topic(dtu_config['customer_id'], 'dtu', dtu_config['dtu_id'], 'status'),
  3854. 'control': build_dtu_topic(dtu_config['customer_id'], 'dtu', dtu_config['dtu_id'], 'control'),
  3855. 'response': build_dtu_topic(dtu_config['customer_id'], 'dtu', dtu_config['dtu_id'], 'response'),
  3856. 'event': build_dtu_topic(dtu_config['customer_id'], 'dtu', dtu_config['dtu_id'], 'event'),
  3857. 'alarm': build_dtu_topic(dtu_config['customer_id'], 'dtu', dtu_config['dtu_id'], 'alarm')
  3858. }
  3859. }
  3860. return jsonify({'success': True, 'data': status})
  3861. except Exception as e:
  3862. logger.error(f"获取DTU状态失败: {str(e)}")
  3863. return jsonify({'success': False, 'message': str(e)}), 500
  3864. @app.route('/api/dtu/control', methods=['POST'])
  3865. def dtu_control():
  3866. """发送DTU控制命令(HTTP 入口,复用 dtu_handle_control 逻辑)"""
  3867. try:
  3868. data = request.json or {}
  3869. # 兼容两种入参格式:
  3870. # 1) {"command": "...", "target": "...", "params": {...}} 简洁格式
  3871. # 2) {"payload": {"command": "...", "target": "...", "params": {...}}} MQTT envelope 格式
  3872. inner = data.get('payload') if isinstance(data.get('payload'), dict) else data
  3873. command = inner.get('command')
  3874. # REBOOT 走快速通道立即执行
  3875. if command == 'REBOOT':
  3876. import subprocess
  3877. logger.warning("执行系统重启命令")
  3878. threading.Thread(target=lambda: (
  3879. time.sleep(1),
  3880. subprocess.run(['reboot'], capture_output=True)
  3881. ), daemon=True).start()
  3882. return jsonify({'success': True, 'message': '系统正在重启...'})
  3883. # 其他命令委托给 MQTT 处理器(构造 MQTT envelope 后调用)
  3884. mqtt_envelope = {
  3885. 'msg_id': data.get('msg_id', f"http_{int(time.time() * 1000)}_{uuid.uuid4().hex[:6]}"),
  3886. 'timestamp': data.get('timestamp', int(time.time() * 1000)),
  3887. 'dtu_id': data.get('dtu_id', dtu_config.get('dtu_id')),
  3888. 'type': 'CONTROL',
  3889. 'payload': {
  3890. 'command': command,
  3891. 'target': inner.get('target'),
  3892. 'params': inner.get('params', {})
  3893. }
  3894. }
  3895. # 调用 MQTT 处理器(其内部会发送响应到 dtu/.../response 主题)
  3896. dtu_handle_control('http/control', mqtt_envelope)
  3897. # 如果命令是 QUERY_JUMPER_STATUS / QUERY_ENV_SENSOR / QUERY_DTU_STATUS,
  3898. # 实际响应已发布到对应主题,HTTP 端点仅返回成功标记
  3899. if command in ('QUERY_JUMPER_STATUS', 'QUERY_ENV_SENSOR', 'QUERY_DTU_STATUS'):
  3900. return jsonify({
  3901. 'success': True,
  3902. 'message': f'{command} 已发布到 MQTT 主题'
  3903. })
  3904. # 其他命令: 从 MQTT 主题同步拉取最新响应(简化处理:直接返回已发布标记)
  3905. return jsonify({
  3906. 'success': True,
  3907. 'message': f'{command} 命令已下发,等待 MQTT 响应'
  3908. })
  3909. except Exception as e:
  3910. logger.error(f"DTU控制命令失败: {str(e)}")
  3911. return jsonify({'success': False, 'message': str(e)}), 500
  3912. # OTA状态存储
  3913. ota_status = {
  3914. 'status': 'IDLE', # IDLE, DOWNLOADING, VERIFYING, FLASHING, SUCCESS, FAILED
  3915. 'progress': 0,
  3916. 'firmware_version': None,
  3917. 'target_version': None,
  3918. 'error_code': None,
  3919. 'error_message': None,
  3920. 'last_update': None
  3921. }
  3922. @app.route('/api/dtu/ota_detect', methods=['POST'])
  3923. def ota_detect():
  3924. """检测固件包信息(自动下载并解析 manifest)"""
  3925. try:
  3926. url = request.json.get('url', '')
  3927. if not url:
  3928. return jsonify({'success': False, 'message': '请提供固件 URL'}), 400
  3929. import urllib.request, tempfile, tarfile, json, hashlib
  3930. tmp = tempfile.mktemp(suffix='.tar.gz')
  3931. try:
  3932. urllib.request.urlretrieve(url, tmp)
  3933. except Exception as e:
  3934. return jsonify({'success': False, 'message': f'下载失败: {str(e)}'}), 400
  3935. file_size = os.path.getsize(tmp)
  3936. h = hashlib.md5()
  3937. with open(tmp, 'rb') as f:
  3938. for chunk in iter(lambda: f.read(65536), b''):
  3939. h.update(chunk)
  3940. md5sum = h.hexdigest()
  3941. version = ''
  3942. try:
  3943. with tarfile.open(tmp, 'r:gz') as tar:
  3944. m = tar.extractfile('firmware/firmware.json')
  3945. if m:
  3946. manifest = json.loads(m.read())
  3947. version = manifest.get('version', '')
  3948. except Exception as e:
  3949. logger.warning(f"解析 firmware.json 失败: {e}")
  3950. os.remove(tmp)
  3951. return jsonify({
  3952. 'success': True,
  3953. 'data': {
  3954. 'file_size': file_size,
  3955. 'checksum': md5sum,
  3956. 'checksum_type': 'MD5',
  3957. 'firmware_version': version
  3958. }
  3959. })
  3960. except Exception as e:
  3961. logger.error(f"检测固件失败: {str(e)}")
  3962. return jsonify({'success': False, 'message': str(e)}), 500
  3963. @app.route('/api/dtu/ota_status', methods=['GET'])
  3964. def get_ota_status():
  3965. """获取OTA升级状态"""
  3966. try:
  3967. return jsonify({
  3968. 'success': True,
  3969. 'data': {
  3970. 'current_firmware': dtu_config.get('firmware_version', 'v1.0.0'),
  3971. 'ota_status': ota_status.get('status', 'IDLE'),
  3972. 'ota_progress': ota_status.get('progress', 0),
  3973. 'target_version': ota_status.get('target_version'),
  3974. 'error_code': ota_status.get('error_code'),
  3975. 'error_message': ota_status.get('error_message'),
  3976. 'last_update': ota_status.get('last_update')
  3977. }
  3978. })
  3979. except Exception as e:
  3980. logger.error(f"获取OTA状态失败: {str(e)}")
  3981. return jsonify({'success': False, 'message': str(e)}), 500
  3982. def handle_ota_status(dtu_id, payload):
  3983. """处理MQTT上报的OTA状态"""
  3984. ota_status['status'] = payload.get('ota_status', 'IDLE')
  3985. ota_status['progress'] = payload.get('ota_progress', 0)
  3986. ota_status['firmware_version'] = payload.get('firmware_version')
  3987. ota_status['last_update'] = time.strftime('%Y-%m-%d %H:%M:%S')
  3988. error_code = payload.get('error_code')
  3989. if error_code and error_code != 0:
  3990. ota_status['error_code'] = error_code
  3991. ota_status['error_message'] = payload.get('error_message', get_ota_error_message(error_code))
  3992. ota_status['status'] = 'FAILED'
  3993. elif ota_status['status'] == 'SUCCESS':
  3994. ota_status['error_code'] = None
  3995. ota_status['error_message'] = None
  3996. if ota_status.get('firmware_version'):
  3997. dtu_config['firmware_version'] = ota_status['firmware_version']
  3998. save_dtu_config()
  3999. logger.info(f"MQTT OTA状态更新: {ota_status['status']}, 进度: {ota_status['progress']}%")
  4000. def get_ota_error_message(error_code):
  4001. error_messages = {1021: '已是目标版本', 1022: '校验失败', 1023: '下载失败', 1024: '写入失败', 1025: '存储空间不足'}
  4002. return error_messages.get(error_code, f'未知错误码: {error_code}')
  4003. def _publish_ota_progress():
  4004. """通过MQTT发布OTA进度"""
  4005. try:
  4006. if not _mqtt_connected():
  4007. return
  4008. topic = build_dtu_topic(dtu_config['customer_id'], 'dtu', dtu_config['dtu_id'], 'status')
  4009. payload = {
  4010. 'msg_id': f"ota_status_{int(time.time() * 1000)}",
  4011. 'timestamp': int(time.time() * 1000),
  4012. 'dtu_id': dtu_config['dtu_id'],
  4013. 'type': 'STATUS',
  4014. 'payload': {
  4015. 'ota_status': ota_status.get('status', 'IDLE'),
  4016. 'ota_progress': ota_status.get('progress', 0),
  4017. 'firmware_version': ota_status.get('target_version', ''),
  4018. 'error_code': ota_status.get('error_code', 0),
  4019. 'error_message': ota_status.get('error_message')
  4020. }
  4021. }
  4022. mqtt_client.publish(topic, json.dumps(payload), qos=1)
  4023. except Exception:
  4024. pass
  4025. def _send_ota_error_response(error_code, original_msg_id=''):
  4026. """通过MQTT发送OTA错误响应(1021-1025)"""
  4027. try:
  4028. if not _mqtt_connected():
  4029. return
  4030. topic = build_dtu_topic(dtu_config['customer_id'], 'dtu', dtu_config['dtu_id'], 'response')
  4031. payload = {
  4032. 'msg_id': f"rsp_ota_{int(time.time() * 1000)}_{uuid.uuid4().hex[:6]}",
  4033. 'timestamp': int(time.time() * 1000),
  4034. 'dtu_id': dtu_config['dtu_id'],
  4035. 'type': 'RESPONSE',
  4036. 'payload': {
  4037. 'original_msg_id': original_msg_id,
  4038. 'command': 'OTA_UPGRADE',
  4039. 'success': False,
  4040. 'result': None,
  4041. 'error_code': error_code,
  4042. 'error_message': get_ota_error_message(error_code)
  4043. }
  4044. }
  4045. mqtt_client.publish(topic, json.dumps(payload), qos=1)
  4046. except Exception:
  4047. pass
  4048. def _run_ota(params):
  4049. """执行OTA升级(后台线程,供HTTP和MQTT共用)"""
  4050. import subprocess, os, hashlib, shutil, tarfile, tempfile, urllib.request
  4051. url = params['firmware_url']
  4052. version = params['firmware_version']
  4053. file_size = params['file_size']
  4054. checksum = params['checksum']
  4055. checksum_type = params.get('checksum_type', 'MD5').upper()
  4056. force = params.get('force_upgrade', False)
  4057. ota_status['status'] = 'DOWNLOADING'
  4058. ota_status['progress'] = 0
  4059. ota_status['target_version'] = version
  4060. ota_status['error_code'] = None
  4061. ota_status['error_message'] = None
  4062. ota_status['last_update'] = time.strftime('%Y-%m-%d %H:%M:%S')
  4063. _publish_ota_progress()
  4064. try:
  4065. tmp_dir = tempfile.mkdtemp(prefix='ota_')
  4066. fw_path = os.path.join(tmp_dir, 'firmware.tar.gz')
  4067. ota_status['progress'] = 5
  4068. ota_status['last_update'] = time.strftime('%Y-%m-%d %H:%M:%S')
  4069. logger.info(f"OTA: 开始下载固件 {url}")
  4070. req = urllib.request.Request(url, headers={'User-Agent': 'OTA-Updater'})
  4071. # 进度上报节流: 每 10% 或每 5 秒
  4072. last_published_pct = 0
  4073. last_publish_time = time.time()
  4074. with urllib.request.urlopen(req, timeout=120) as resp:
  4075. with open(fw_path, 'wb') as f:
  4076. total = int(resp.headers.get('Content-Length', 0))
  4077. downloaded = 0
  4078. while True:
  4079. chunk = resp.read(65536)
  4080. if not chunk: break
  4081. f.write(chunk)
  4082. downloaded += len(chunk)
  4083. if total:
  4084. pct = 5 + int(downloaded / total * 30)
  4085. ota_status['progress'] = min(pct, 35)
  4086. now = time.time()
  4087. if pct - last_published_pct >= 10 or (now - last_publish_time) >= 5:
  4088. last_published_pct = pct
  4089. last_publish_time = now
  4090. ota_status['last_update'] = time.strftime('%Y-%m-%d %H:%M:%S')
  4091. _publish_ota_progress()
  4092. dl_size = os.path.getsize(fw_path)
  4093. ota_status['progress'] = 40
  4094. ota_status['last_update'] = time.strftime('%Y-%m-%d %H:%M:%S')
  4095. _publish_ota_progress()
  4096. if abs(dl_size - file_size) > 1024:
  4097. raise Exception(f"文件大小不匹配: 预期{file_size}, 实际{dl_size}")
  4098. ota_status['status'] = 'VERIFYING'
  4099. ota_status['progress'] = 50
  4100. ota_status['last_update'] = time.strftime('%Y-%m-%d %H:%M:%S')
  4101. _publish_ota_progress()
  4102. h = hashlib.new(checksum_type)
  4103. with open(fw_path, 'rb') as f:
  4104. for chunk in iter(lambda: f.read(65536), b''): h.update(chunk)
  4105. actual_checksum = h.hexdigest().lower()
  4106. if actual_checksum != checksum.lower():
  4107. raise Exception(f"校验和不匹配: 预期{checksum}, 实际{actual_checksum}")
  4108. ota_status['progress'] = 60
  4109. ota_status['last_update'] = time.strftime('%Y-%m-%d %H:%M:%S')
  4110. _publish_ota_progress()
  4111. extract_dir = os.path.join(tmp_dir, 'firmware')
  4112. os.makedirs(extract_dir, exist_ok=True)
  4113. with tarfile.open(fw_path, 'r:gz') as tar: tar.extractall(extract_dir)
  4114. manifest_path = os.path.join(extract_dir, 'firmware', 'firmware.json')
  4115. if not os.path.exists(manifest_path):
  4116. raise Exception("固件包缺少 firmware.json")
  4117. with open(manifest_path, 'r') as f: manifest = json.load(f)
  4118. fw_version = manifest.get('version', version)
  4119. if not force:
  4120. current_ver = dtu_config.get('firmware_version', 'v0.0.0')
  4121. if fw_version == current_ver:
  4122. raise Exception(f'已是目标版本 {current_ver}')
  4123. ota_status['status'] = 'FLASHING'
  4124. ota_status['progress'] = 70
  4125. ota_status['last_update'] = time.strftime('%Y-%m-%d %H:%M:%S')
  4126. _publish_ota_progress()
  4127. fw_root = os.path.join(extract_dir, 'firmware')
  4128. project_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
  4129. excludes = {'__pycache__', '.git', 'log', 'venv'}
  4130. for root, dirs, files in os.walk(fw_root):
  4131. rel = os.path.relpath(root, fw_root)
  4132. if rel == '.': rel = ''
  4133. parts = rel.split(os.sep) if rel else []
  4134. if parts and parts[0] in excludes: continue
  4135. for fname in files:
  4136. if fname == 'firmware.json': continue
  4137. src = os.path.join(root, fname)
  4138. dst = os.path.join(project_dir, rel, fname)
  4139. os.makedirs(os.path.dirname(dst), exist_ok=True)
  4140. shutil.copy2(src, dst)
  4141. ota_status['progress'] = 90
  4142. ota_status['last_update'] = time.strftime('%Y-%m-%d %H:%M:%S')
  4143. dtu_config['firmware_version'] = fw_version
  4144. save_dtu_config()
  4145. ota_status['status'] = 'SUCCESS'
  4146. ota_status['progress'] = 100
  4147. ota_status['firmware_version'] = fw_version
  4148. ota_status['last_update'] = time.strftime('%Y-%m-%d %H:%M:%S')
  4149. _publish_ota_progress()
  4150. shutil.rmtree(tmp_dir, ignore_errors=True)
  4151. for i in range(10, 0, -1):
  4152. ota_status['last_update'] = time.strftime('%Y-%m-%d %H:%M:%S')
  4153. time.sleep(1)
  4154. logger.info("OTA: 重启服务...")
  4155. subprocess.Popen(
  4156. [subprocess.sys.executable, '-m', 'flask', 'run', '--host=0.0.0.0', '--port=5001'],
  4157. cwd=os.path.dirname(os.path.abspath(__file__)),
  4158. stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL
  4159. )
  4160. os._exit(0)
  4161. except Exception as e:
  4162. ota_status['status'] = 'FAILED'
  4163. ota_status['error_message'] = str(e)
  4164. ota_status['last_update'] = time.strftime('%Y-%m-%d %H:%M:%S')
  4165. # 映射错误到标准错误码 (1021-1025)
  4166. msg = str(e)
  4167. if '校验和' in msg or 'firmware.json' in msg:
  4168. ota_status['error_code'] = 1022
  4169. elif '已是目标版本' in msg:
  4170. ota_status['error_code'] = 1021
  4171. elif '下载' in msg or 'urlopen' in msg or 'Connection' in msg or 'timeout' in msg:
  4172. ota_status['error_code'] = 1023
  4173. elif '磁盘' in msg or '空间' in msg or 'No space' in msg:
  4174. ota_status['error_code'] = 1025
  4175. elif '写入' in msg or 'flash' in msg.lower() or 'shutil' in msg:
  4176. ota_status['error_code'] = 1024
  4177. else:
  4178. ota_status['error_code'] = 9999
  4179. _publish_ota_progress()
  4180. # 通过MQTT发送错误响应(同时适配HTTP和MQTT触发)
  4181. _send_ota_error_response(ota_status['error_code'], params.get('_incoming_msg_id', ''))
  4182. logger.error(f"OTA: 升级失败 - {str(e)}")
  4183. if 'tmp_dir' in dir() and tmp_dir and os.path.exists(tmp_dir):
  4184. shutil.rmtree(tmp_dir, ignore_errors=True)
  4185. @app.route('/api/dtu/ota_upgrade', methods=['POST'])
  4186. def trigger_ota_upgrade():
  4187. """触发OTA升级"""
  4188. try:
  4189. data = request.json
  4190. if not data:
  4191. return jsonify({'success': False, 'message': '请求体不能为空'}), 400
  4192. required_fields = ['firmware_url', 'firmware_version', 'file_size', 'checksum', 'checksum_type']
  4193. missing = [f for f in required_fields if not data.get(f)]
  4194. if missing:
  4195. return jsonify({'success': False, 'message': f'缺少必填字段: {", ".join(missing)}'}), 400
  4196. # 已是目标版本且未强制: 1021 (同时通过 MQTT 上报响应)
  4197. force_upgrade = data.get('force_upgrade', False)
  4198. current_version = dtu_config.get('firmware_version', 'v1.0.0')
  4199. if data['firmware_version'] == current_version and not force_upgrade:
  4200. # 通过 MQTT 上报 1021 响应 (与其他错误码一致)
  4201. try:
  4202. _send_ota_error_response(1021, data.get('msg_id', ''))
  4203. except Exception:
  4204. pass
  4205. return jsonify({
  4206. 'success': False,
  4207. 'error_code': 1021,
  4208. 'message': f'已是目标版本: {current_version}'
  4209. }), 400
  4210. # 注入 _incoming_msg_id 使 MQTT 错误响应可回填
  4211. ota_params = {**data, '_incoming_msg_id': data.get('msg_id', '')}
  4212. t = threading.Thread(target=_run_ota, args=(ota_params,), daemon=True)
  4213. t.start()
  4214. return jsonify({
  4215. 'success': True,
  4216. 'message': 'OTA升级已启动',
  4217. 'data': {
  4218. 'target_version': data['firmware_version'],
  4219. 'ota_status': 'DOWNLOADING'
  4220. }
  4221. })
  4222. except Exception as e:
  4223. logger.error(f"触发OTA升级失败: {str(e)}")
  4224. return jsonify({'success': False, 'message': str(e)}), 500
  4225. # ==================== 环境传感器 API ====================
  4226. # 环境传感器数据存储
  4227. env_sensor_data = {
  4228. 'temperature': None, # 环境温度 (℃)
  4229. 'humidity': None, # 环境湿度 (%)
  4230. 'dtu_temperature': None, # DTU主板温度 (℃)
  4231. 'update_time': None, # 数据更新时间
  4232. 'sensor_update_time': None, # 传感器更新时间
  4233. 'connected': False # 传感器连接状态
  4234. }
  4235. # 环境传感器告警阈值
  4236. env_sensor_threshold = {
  4237. 'temp_high': 45.0, # 环境温度上限 (℃)
  4238. 'temp_low': -10.0, # 环境温度下限 (℃)
  4239. 'humidity_high': 80.0, # 环境湿度上限 (%)
  4240. 'humidity_low': 20.0, # 环境湿度下限 (%)
  4241. 'dtu_temp_high': 70.0 # DTU主板温度上限 (℃)
  4242. }
  4243. # 环境传感器历史数据 (保留最近1000条)
  4244. env_sensor_history = []
  4245. @app.route('/api/sensor/env', methods=['GET'])
  4246. def get_env_sensor_data():
  4247. """获取环境传感器当前数据"""
  4248. data = dict(env_sensor_data)
  4249. # DTU 主板温度回退: 若外部未上报,从 sysfs 本地读取
  4250. if data.get('dtu_temperature') is None:
  4251. try:
  4252. import psutil
  4253. temps = psutil.sensors_temperatures(fahrenheit=False) or {}
  4254. for chip_name, entries in temps.items():
  4255. for entry in entries:
  4256. if entry.current and entry.current > 0:
  4257. data['dtu_temperature'] = round(entry.current, 1)
  4258. break
  4259. if data['dtu_temperature'] is not None:
  4260. break
  4261. except Exception:
  4262. pass
  4263. if data.get('dtu_temperature') is None:
  4264. import glob as _glob
  4265. for tz in sorted(_glob.glob('/sys/class/thermal/thermal_zone*/temp')):
  4266. try:
  4267. with open(tz) as f:
  4268. v = int(f.read().strip())
  4269. if v > 0:
  4270. data['dtu_temperature'] = round(v / 1000.0, 1)
  4271. data['update_time'] = data.get('update_time') or __import__('datetime').datetime.now().strftime('%Y-%m-%d %H:%M:%S')
  4272. break
  4273. except Exception:
  4274. continue
  4275. # 附加 CPU/内存使用率(始终本地读取)
  4276. data['cpu_usage'] = _get_cpu_usage()
  4277. data['memory_usage'] = _get_memory_usage()
  4278. return jsonify({
  4279. 'success': True,
  4280. 'data': data
  4281. })
  4282. @app.route('/api/sensor/threshold', methods=['GET', 'POST'])
  4283. def handle_env_sensor_threshold():
  4284. """获取或设置环境传感器告警阈值"""
  4285. if request.method == 'GET':
  4286. return jsonify({
  4287. 'success': True,
  4288. 'data': env_sensor_threshold
  4289. })
  4290. else:
  4291. try:
  4292. data = request.get_json()
  4293. for key in env_sensor_threshold.keys():
  4294. if key in data:
  4295. env_sensor_threshold[key] = float(data[key])
  4296. logger.info(f"更新环境传感器告警阈值: {env_sensor_threshold}")
  4297. return jsonify({
  4298. 'success': True,
  4299. 'message': '阈值设置成功',
  4300. 'data': env_sensor_threshold
  4301. })
  4302. except Exception as e:
  4303. logger.error(f"设置告警阈值失败: {str(e)}")
  4304. return jsonify({'success': False, 'message': str(e)}), 400
  4305. @app.route('/api/sensor/history', methods=['GET'])
  4306. def get_env_sensor_history():
  4307. """获取环境传感器历史数据"""
  4308. try:
  4309. import datetime
  4310. # 获取查询参数
  4311. range_param = request.args.get('range', '24h')
  4312. start_time = request.args.get('start_time')
  4313. end_time = request.args.get('end_time')
  4314. limit = request.args.get('limit', 100, type=int)
  4315. filtered_history = env_sensor_history
  4316. # 解析range参数
  4317. if not start_time and not end_time:
  4318. now = datetime.datetime.now()
  4319. if range_param == '1h':
  4320. start_time = (now - datetime.timedelta(hours=1)).strftime('%Y-%m-%d %H:%M:%S')
  4321. elif range_param == '6h':
  4322. start_time = (now - datetime.timedelta(hours=6)).strftime('%Y-%m-%d %H:%M:%S')
  4323. elif range_param == '24h':
  4324. start_time = (now - datetime.timedelta(hours=24)).strftime('%Y-%m-%d %H:%M:%S')
  4325. elif range_param == '7d':
  4326. start_time = (now - datetime.timedelta(days=7)).strftime('%Y-%m-%d %H:%M:%S')
  4327. elif range_param == '30d':
  4328. start_time = (now - datetime.timedelta(days=30)).strftime('%Y-%m-%d %H:%M:%S')
  4329. # 按时间过滤
  4330. if start_time:
  4331. filtered_history = [h for h in filtered_history if h.get('update_time') >= start_time]
  4332. if end_time:
  4333. filtered_history = [h for h in filtered_history if h.get('update_time') <= end_time]
  4334. # 按时间排序 (新到旧)
  4335. filtered_history = sorted(filtered_history, key=lambda x: x.get('update_time', ''), reverse=True)
  4336. # 限制返回数量
  4337. filtered_history = filtered_history[:limit]
  4338. return jsonify({
  4339. 'success': True,
  4340. 'data': filtered_history,
  4341. 'total': len(filtered_history)
  4342. })
  4343. except Exception as e:
  4344. logger.error(f"获取历史数据失败: {str(e)}")
  4345. return jsonify({'success': False, 'message': str(e)}), 500
  4346. def update_env_sensor_data_full(temperature, humidity, dtu_temperature, sensor_update_time):
  4347. """更新环境传感器数据 (由MQTT消息触发) - 完整版,含历史和告警"""
  4348. import datetime
  4349. now = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
  4350. env_sensor_data['temperature'] = temperature
  4351. env_sensor_data['humidity'] = humidity
  4352. env_sensor_data['dtu_temperature'] = dtu_temperature
  4353. env_sensor_data['sensor_update_time'] = sensor_update_time
  4354. env_sensor_data['update_time'] = now
  4355. env_sensor_data['connected'] = True
  4356. # 同步到 MQTT 上行所用的 _env_sensor_data
  4357. _env_sensor_data['temperature'] = temperature
  4358. _env_sensor_data['humidity'] = humidity
  4359. _env_sensor_data['sensor_update_time'] = sensor_update_time
  4360. # 添加到历史记录
  4361. history_record = {
  4362. 'temperature': temperature,
  4363. 'humidity': humidity,
  4364. 'dtu_temperature': dtu_temperature,
  4365. 'sensor_update_time': sensor_update_time,
  4366. 'update_time': now
  4367. }
  4368. env_sensor_history.append(history_record)
  4369. # 保留最近1000条
  4370. if len(env_sensor_history) > 1000:
  4371. env_sensor_history[:] = env_sensor_history[-1000:]
  4372. # 检查告警
  4373. check_env_sensor_alarms(temperature, humidity, dtu_temperature)
  4374. logger.info(f"环境传感器数据更新: 温度={temperature}℃, 湿度={humidity}%, DTU温度={dtu_temperature}℃")
  4375. def dht11_data_callback(temperature, humidity):
  4376. """DHT11 本地 GPIO 传感器数据回调。
  4377. 将读取到的温湿度更新到环境传感器数据区,并通过 MQTT 上报。
  4378. 保留已有的 dtu_temperature(主板温度),仅更新环境温湿度字段。
  4379. 若外部未上报 dtu_temperature,则从本地 sysfs 读取(与 /api/sensor/env 回退逻辑一致),
  4380. 确保历史记录中 DTU 温度字段有值。
  4381. """
  4382. try:
  4383. sensor_update_time = int(time.time() * 1000)
  4384. dtu_temperature = env_sensor_data.get('dtu_temperature')
  4385. # DTU 主板温度回退:外部未上报时从 sysfs 本地读取,并回填 env_sensor_data
  4386. # 否则历史记录 dtu_temperature 会一直是 None(/api/sensor/env 有回退但历史写入没有)
  4387. if dtu_temperature is None:
  4388. try:
  4389. import psutil
  4390. temps = psutil.sensors_temperatures(fahrenheit=False) or {}
  4391. for chip_name, entries in temps.items():
  4392. for entry in entries:
  4393. if entry.current and entry.current > 0:
  4394. dtu_temperature = round(entry.current, 1)
  4395. break
  4396. if dtu_temperature is not None:
  4397. break
  4398. except Exception:
  4399. pass
  4400. if dtu_temperature is None:
  4401. import glob as _glob
  4402. for tz in sorted(_glob.glob('/sys/class/thermal/thermal_zone*/temp')):
  4403. try:
  4404. with open(tz) as f:
  4405. v = int(f.read().strip())
  4406. if v > 0:
  4407. dtu_temperature = round(v / 1000.0, 1)
  4408. break
  4409. except Exception:
  4410. continue
  4411. # 回填到 env_sensor_data,避免每次回调都重新读 sysfs
  4412. if dtu_temperature is not None:
  4413. env_sensor_data['dtu_temperature'] = dtu_temperature
  4414. update_env_sensor_data_full(temperature, humidity, dtu_temperature, sensor_update_time)
  4415. # 立即通过 MQTT 上报环境传感器数据
  4416. try:
  4417. dtu_publish_env_sensor()
  4418. except Exception as pub_err:
  4419. logger.warning(f"DHT11 数据 MQTT 上报失败: {pub_err}")
  4420. logger.info(f"DHT11 本地传感器数据已处理: 温度={temperature}°C, 湿度={humidity}%")
  4421. except Exception as e:
  4422. logger.error(f"处理 DHT11 本地传感器数据失败: {e}")
  4423. def check_env_sensor_alarms(temperature, humidity, dtu_temperature):
  4424. """检查环境传感器告警"""
  4425. alarms = []
  4426. if temperature is not None:
  4427. if temperature > env_sensor_threshold['temp_high']:
  4428. alarms.append({
  4429. 'type': 'temperature_high',
  4430. 'message': f'环境温度过高: {temperature}℃ (阈值: {env_sensor_threshold["temp_high"]}℃)',
  4431. 'level': 'warning'
  4432. })
  4433. elif temperature < env_sensor_threshold['temp_low']:
  4434. alarms.append({
  4435. 'type': 'temperature_low',
  4436. 'message': f'环境温度过低: {temperature}℃ (阈值: {env_sensor_threshold["temp_low"]}℃)',
  4437. 'level': 'warning'
  4438. })
  4439. if humidity is not None:
  4440. if humidity > env_sensor_threshold['humidity_high']:
  4441. alarms.append({
  4442. 'type': 'humidity_high',
  4443. 'message': f'环境湿度过高: {humidity}% (阈值: {env_sensor_threshold["humidity_high"]}%)',
  4444. 'level': 'warning'
  4445. })
  4446. elif humidity < env_sensor_threshold['humidity_low']:
  4447. alarms.append({
  4448. 'type': 'humidity_low',
  4449. 'message': f'环境湿度过低: {humidity}% (阈值: {env_sensor_threshold["humidity_low"]}%)',
  4450. 'level': 'warning'
  4451. })
  4452. if dtu_temperature is not None:
  4453. if dtu_temperature > env_sensor_threshold['dtu_temp_high']:
  4454. alarms.append({
  4455. 'type': 'dtu_temp_high',
  4456. 'message': f'DTU主板温度过高: {dtu_temperature}℃ (阈值: {env_sensor_threshold["dtu_temp_high"]}℃)',
  4457. 'level': 'critical'
  4458. })
  4459. # 发送告警通知
  4460. for alarm in alarms:
  4461. logger.warning(f"环境传感器告警: {alarm['message']}")
  4462. # 可以通过WebSocket发送告警
  4463. socketio.emit('env_sensor_alarm', alarm)
  4464. # ========== 运行数据备份 / 恢复 ==========
  4465. SYSTEM_ENV_FILE = '/etc/default/dzxj_dtu'
  4466. _SNAPSHOT_TARGETS = {
  4467. 'dtu_config': DTU_CONFIG_FILE,
  4468. 'serial_config': SERIAL_CONFIG_FILE,
  4469. 'mqtt_config': MQTT_CONFIG_FILE,
  4470. 'devices': DEVICE_CONFIG_FILE,
  4471. }
  4472. def _read_system_env():
  4473. """读取设备环境变量文件(/etc/default/dzxj_dtu),不存在返回 None"""
  4474. try:
  4475. if os.path.exists(SYSTEM_ENV_FILE):
  4476. with open(SYSTEM_ENV_FILE, 'r', encoding='utf-8') as f:
  4477. return f.read()
  4478. except Exception as e:
  4479. logger.warning(f"读取环境变量文件失败: {e}")
  4480. return None
  4481. def _collect_backup_files():
  4482. """收集备份内容:4 个配置文件 + 系统环境变量文件(可选)"""
  4483. files = {}
  4484. dtu_data = None
  4485. try:
  4486. if os.path.exists(DTU_CONFIG_FILE):
  4487. dtu_data = config_backup.read_json_file(DTU_CONFIG_FILE)
  4488. except Exception as e:
  4489. logger.warning(f"读取 dtu_config.json 失败,改用内存配置: {e}")
  4490. files['dtu_config'] = dtu_data if isinstance(dtu_data, dict) else dict(dtu_config)
  4491. for key, path in (('serial_config', SERIAL_CONFIG_FILE),
  4492. ('mqtt_config', MQTT_CONFIG_FILE),
  4493. ('devices', DEVICE_CONFIG_FILE)):
  4494. data = None
  4495. try:
  4496. if os.path.exists(path):
  4497. data = config_backup.read_json_file(path)
  4498. except Exception as e:
  4499. logger.warning(f"读取 {path} 失败: {e}")
  4500. files[key] = data
  4501. if not isinstance(files.get('devices'), dict):
  4502. files['devices'] = address_config.get_stored_devices()
  4503. files['system_env'] = _read_system_env()
  4504. return files
  4505. def _configs_differ(a, b):
  4506. try:
  4507. return (json.dumps(a or {}, sort_keys=True, ensure_ascii=False)
  4508. != json.dumps(b or {}, sort_keys=True, ensure_ascii=False))
  4509. except Exception:
  4510. return True
  4511. def _backup_preview(bundle):
  4512. """备份包预览信息(导入前确认用)"""
  4513. files = bundle.get('files') or {}
  4514. dtu_data = files.get('dtu_config') or {}
  4515. included = {}
  4516. for key, val in files.items():
  4517. if val in (None, '', {}):
  4518. continue
  4519. if isinstance(val, str):
  4520. included[key] = f'{len(val)} 字节文本'
  4521. else:
  4522. included[key] = f'{len(json.dumps(val, ensure_ascii=False))} 字节'
  4523. prefix = bundle.get('effective_uid_prefix')
  4524. return {
  4525. 'created_at': bundle.get('created_at'),
  4526. 'device': bundle.get('device') or {},
  4527. 'files': included,
  4528. 'effective_uid_prefix': prefix,
  4529. 'current_dtu_id': dtu_config.get('dtu_id'),
  4530. 'backup_dtu_id': dtu_data.get('dtu_id'),
  4531. 'will_fix_uid_prefix': bool(prefix) and not dtu_data.get('uid_prefix'),
  4532. 'restart_required': isinstance(files.get('system_env'), str) and bool((files.get('system_env') or '').strip()),
  4533. 'devices_count': len(files.get('devices') or {}),
  4534. 'port_mapping_panels': len(dtu_data.get('port_mappings') or {}),
  4535. }
  4536. def _apply_backup_bundle(bundle):
  4537. """应用备份包:写文件 + 内存生效 + 自动重连,返回结果摘要"""
  4538. files = bundle.get('files') or {}
  4539. applied = []
  4540. errors = []
  4541. # 1) dtu_config(固化 effective_uid_prefix,保证换机后卡号前缀一致)
  4542. dtu_data = files.get('dtu_config')
  4543. if isinstance(dtu_data, dict):
  4544. dtu_data = dict(dtu_data)
  4545. prefix = (bundle.get('effective_uid_prefix') or '').strip().upper()
  4546. if prefix and not dtu_data.get('uid_prefix'):
  4547. dtu_data['uid_prefix'] = prefix
  4548. applied.append(f'已固化 UID 前缀: {prefix}')
  4549. dtu_config.update(dtu_data)
  4550. save_dtu_config()
  4551. applied.append('dtu_config')
  4552. # 2) 串口配置(变化则重连)
  4553. sc = files.get('serial_config')
  4554. if isinstance(sc, dict):
  4555. old_sc = load_serial_config()
  4556. if _save_config_file(SERIAL_CONFIG_FILE, sc, '串口配置'):
  4557. applied.append('serial_config')
  4558. if _configs_differ(old_sc, sc):
  4559. try:
  4560. if _serial_connected():
  4561. serial_client.disconnect()
  4562. auto_connect_serial()
  4563. except Exception as e:
  4564. errors.append(f'串口重连失败: {e}')
  4565. # 3) MQTT 配置(变化则重连)
  4566. mc = files.get('mqtt_config')
  4567. if isinstance(mc, dict):
  4568. old_mc = load_mqtt_config()
  4569. if _save_config_file(MQTT_CONFIG_FILE, mc, 'MQTT配置'):
  4570. applied.append('mqtt_config')
  4571. if _configs_differ(old_mc, mc):
  4572. try:
  4573. if _mqtt_connected():
  4574. mqtt_client.disconnect()
  4575. auto_connect_mqtt()
  4576. except Exception as e:
  4577. errors.append(f'MQTT重连失败: {e}')
  4578. # 4) 线架绑定表(重建面板配置)
  4579. dev = files.get('devices')
  4580. if isinstance(dev, dict):
  4581. if _save_config_file(DEVICE_CONFIG_FILE, dev, '设备配置'):
  4582. applied.append('devices')
  4583. address_config.stored_devices = {str(k).lower(): int(v) for k, v in dev.items()}
  4584. panel_config.clear()
  4585. port_state.clear()
  4586. panel_fail_count.clear()
  4587. sync_panel_config()
  4588. # 5) 系统环境变量文件(需重启服务生效)
  4589. restart_required = False
  4590. env_text = files.get('system_env')
  4591. if isinstance(env_text, str) and env_text.strip():
  4592. try:
  4593. with open(SYSTEM_ENV_FILE, 'w', encoding='utf-8') as f:
  4594. f.write(env_text)
  4595. applied.append('system_env')
  4596. restart_required = True
  4597. except Exception as e:
  4598. errors.append(f'写入环境变量文件失败(需 root): {e}')
  4599. # 6) 后台补一次注册上报
  4600. def _post_restore_refresh():
  4601. try:
  4602. if _mqtt_connected():
  4603. dtu_register()
  4604. except Exception as e:
  4605. logger.warning(f"恢复后注册上报失败: {e}")
  4606. threading.Thread(target=_post_restore_refresh, daemon=True).start()
  4607. return {'applied': applied, 'errors': errors, 'restart_required': restart_required}
  4608. @app.route('/api/backup/export', methods=['GET'])
  4609. def backup_export():
  4610. """导出整机运行数据备份包(download=1 时作为文件下载)"""
  4611. try:
  4612. files = _collect_backup_files()
  4613. bundle = config_backup.build_bundle(
  4614. files,
  4615. device_meta={
  4616. 'dtu_id': dtu_config.get('dtu_id'),
  4617. 'customer_id': dtu_config.get('customer_id'),
  4618. 'firmware_version': dtu_config.get('firmware_version'),
  4619. 'hardware_version': dtu_config.get('hardware_version'),
  4620. 'hostname': os.uname().nodename,
  4621. },
  4622. uid_prefix=_dtu_uid_prefix(),
  4623. )
  4624. except Exception as e:
  4625. logger.error(f"导出备份失败: {e}")
  4626. return jsonify({'success': False, 'message': str(e)}), 500
  4627. if request.args.get('download') in ('1', 'true', 'True'):
  4628. text = json.dumps(bundle, ensure_ascii=False, indent=2)
  4629. fname = f"dzxj_dtu_backup_{dtu_config.get('dtu_id', 'dtu')}_{time.strftime('%Y%m%d_%H%M%S')}.json"
  4630. resp = app.response_class(text, mimetype='application/json')
  4631. resp.headers['Content-Disposition'] = f'attachment; filename="{fname}"'
  4632. return resp
  4633. return jsonify({'success': True, 'bundle': bundle})
  4634. @app.route('/api/backup/import', methods=['POST'])
  4635. def backup_import():
  4636. """导入整机备份(支持上传文件/JSON body;dry_run=1 仅预览不应用)"""
  4637. try:
  4638. raw = b''
  4639. if 'file' in request.files:
  4640. raw = request.files['file'].read()
  4641. dry_run = request.args.get('dry_run') in ('1', 'true', 'True')
  4642. if not raw:
  4643. body = request.get_json(silent=True)
  4644. if isinstance(body, dict) and 'bundle' in body:
  4645. raw = json.dumps(body['bundle'], ensure_ascii=False).encode('utf-8')
  4646. dry_run = dry_run or bool(body.get('dry_run'))
  4647. elif body is not None:
  4648. raw = json.dumps(body, ensure_ascii=False).encode('utf-8')
  4649. else:
  4650. raw = request.get_data() or b''
  4651. if not raw:
  4652. return jsonify({'success': False, 'message': '未收到备份内容'}), 400
  4653. if len(raw) > config_backup.MAX_BUNDLE_BYTES:
  4654. return jsonify({'success': False, 'message': '备份文件过大'}), 400
  4655. try:
  4656. bundle = json.loads(raw.decode('utf-8'))
  4657. except Exception:
  4658. return jsonify({'success': False, 'message': '备份内容不是有效 JSON'}), 400
  4659. ok, msg = config_backup.validate_bundle(bundle)
  4660. if not ok:
  4661. return jsonify({'success': False, 'message': msg}), 400
  4662. preview = _backup_preview(bundle)
  4663. if dry_run:
  4664. return jsonify({'success': True, 'dry_run': True, 'preview': preview})
  4665. logger.info(f"开始导入备份: created_at={bundle.get('created_at')} files={list((bundle.get('files') or {}).keys())}")
  4666. summary = _apply_backup_bundle(bundle)
  4667. logger.info(f"备份导入完成: {summary}")
  4668. return jsonify({'success': True, 'message': '备份已恢复', 'preview': preview, **summary})
  4669. except Exception as e:
  4670. logger.error(f"导入备份失败: {e}")
  4671. return jsonify({'success': False, 'message': str(e)}), 500
  4672. @app.route('/api/backup/list', methods=['GET'])
  4673. def backup_list():
  4674. """列出设备内自动快照"""
  4675. return jsonify({'success': True, 'backups': config_backup.list_snapshots(BACKUP_DIR)})
  4676. @app.route('/api/backup/download', methods=['GET'])
  4677. def backup_download():
  4678. """下载指定快照文件"""
  4679. name = request.args.get('name', '')
  4680. data, err = config_backup.read_snapshot(name, BACKUP_DIR)
  4681. if err:
  4682. return jsonify({'success': False, 'message': err}), 404
  4683. text = json.dumps(data, ensure_ascii=False, indent=2)
  4684. resp = app.response_class(text, mimetype='application/json')
  4685. resp.headers['Content-Disposition'] = f'attachment; filename="{os.path.basename(name)}"'
  4686. return resp
  4687. @app.route('/api/backup/restore_snapshot', methods=['POST'])
  4688. def backup_restore_snapshot():
  4689. """从设备内自动快照恢复单个配置文件"""
  4690. try:
  4691. payload = request.json or {}
  4692. name = payload.get('name', '')
  4693. content, err = config_backup.read_snapshot(name, BACKUP_DIR)
  4694. if err:
  4695. return jsonify({'success': False, 'message': err}), 404
  4696. stem = config_backup.snapshot_stem(name)
  4697. if not stem or stem not in _SNAPSHOT_TARGETS:
  4698. return jsonify({'success': False, 'message': f'未知快照类型: {stem}'}), 400
  4699. bundle = config_backup.build_bundle(
  4700. {stem: content},
  4701. device_meta={'source': 'snapshot', 'snapshot': name},
  4702. )
  4703. ok, msg = config_backup.validate_bundle(bundle)
  4704. if not ok:
  4705. return jsonify({'success': False, 'message': msg}), 400
  4706. summary = _apply_backup_bundle(bundle)
  4707. logger.info(f"已从快照 {name} 恢复: {summary}")
  4708. return jsonify({'success': True, 'message': f'已从快照恢复 {stem}', **summary})
  4709. except Exception as e:
  4710. logger.error(f"快照恢复失败: {e}")
  4711. return jsonify({'success': False, 'message': str(e)}), 500
  4712. @app.route('/api/backup/delete', methods=['POST'])
  4713. def backup_delete():
  4714. """删除指定快照"""
  4715. payload = request.json or {}
  4716. ok, msg = config_backup.delete_snapshot(payload.get('name', ''), BACKUP_DIR)
  4717. return (jsonify({'success': True, 'message': msg}) if ok
  4718. else jsonify({'success': False, 'message': msg})), (200 if ok else 404)
  4719. @app.route('/api/system/restart_service', methods=['POST'])
  4720. def restart_service():
  4721. """重启 dzxj_dtu 服务(延时 1 秒执行,让本次响应先返回)"""
  4722. def _do_restart():
  4723. import subprocess
  4724. time.sleep(1.0)
  4725. try:
  4726. subprocess.run(['systemctl', 'restart', 'dzxj_dtu'], timeout=30)
  4727. logger.info("已执行 systemctl restart dzxj_dtu")
  4728. except Exception as e:
  4729. logger.error(f"重启 dzxj_dtu 服务失败: {e}")
  4730. threading.Thread(target=_do_restart, daemon=True).start()
  4731. return jsonify({'success': True, 'message': 'dzxj_dtu 服务将在 1 秒后重启'})
  4732. if __name__ == '__main__':
  4733. try:
  4734. # 启动前的初始化工作
  4735. logger.info('启动串口-MQTT网关服务...')
  4736. logger.info(f"配置信息: 主机={FLASK_HOST}, 端口={FLASK_PORT}, 调试模式={FLASK_DEBUG}")
  4737. # 加载DTU配置
  4738. load_dtu_config()
  4739. # 加载线架绑定表(地址↔UID)
  4740. loaded = load_device_config()
  4741. logger.info(f"已加载 {len(loaded)} 个线架绑定")
  4742. # 自动连接上次使用的串口(RS485),必须先连接才能做地址段扫描
  4743. logger.info("尝试自动连接串口...")
  4744. auto_connect_serial()
  4745. # 自动连接上次使用的MQTT服务器
  4746. logger.info("尝试自动连接MQTT...")
  4747. auto_connect_mqtt()
  4748. # 根据绑定表重建面板列表(真实在线状态由地址段扫描实时刷新)
  4749. try:
  4750. _st = serial_client.get_status()
  4751. if isinstance(_st, dict) and _st.get('connected', False) and dtu_config.get('enabled'):
  4752. sync_panel_config()
  4753. else:
  4754. logger.info("串口未连接或 DTU 未启用,面板列表将在地址段扫描后建立")
  4755. except Exception as e:
  4756. logger.error(f"启动前检查串口状态失败: {e}")
  4757. logger.info(f"当前面板数: {len(panel_config)},地址段: {_address_range()[0]}-{_address_range()[1]}")
  4758. # 启动DTU心跳定时器
  4759. if dtu_config.get('enabled'):
  4760. start_dtu_heartbeat()
  4761. logger.info("启动DTU心跳定时器")
  4762. # 启动线架地址段扫描循环(注册模型:地址段顺序轮询 + UID 绑定 + 换架自动接管)
  4763. def port_poll_loop():
  4764. while True:
  4765. interval = dtu_config.get('poll_interval_ms', 5000) / 1000.0
  4766. time.sleep(max(1, interval))
  4767. try:
  4768. # 地址配置期间暂停扫描,避免总线上同时出现查询帧和配置帧
  4769. if not address_config.config_lock.acquire(blocking=False):
  4770. logger.info("地址配置进行中,本轮扫描跳过")
  4771. continue
  4772. try:
  4773. _st = serial_client.get_status()
  4774. if not (isinstance(_st, dict) and _st.get('connected', False)):
  4775. continue
  4776. if not dtu_config.get('enabled'):
  4777. continue
  4778. online, offline = _sweep_panel_addresses()
  4779. logger.debug(f"地址段扫描完成: 在线={online}, 离线/缺失={offline}")
  4780. dtu_publish_status(force=True)
  4781. finally:
  4782. address_config.config_lock.release()
  4783. except Exception as e:
  4784. logger.error(f"线架地址段扫描异常: {str(e)}")
  4785. t2 = threading.Thread(target=port_poll_loop, daemon=True)
  4786. t2.start()
  4787. logger.info(f"启动线架地址段扫描线程 (顺序轮询 {_address_range()[0]}-{_address_range()[1]})")
  4788. # 启动 DHT11 本地传感器读取线程
  4789. if dht11_sensor is not None:
  4790. dht11_sensor.on_data = dht11_data_callback
  4791. dht11_sensor.start()
  4792. # 连接 Nextion 串口屏
  4793. if NEXTION_ENABLED:
  4794. try:
  4795. ok, msg = screen_display.connect(NEXTION_SCREEN_PORT, NEXTION_SCREEN_BAUD)
  4796. if ok:
  4797. screen_display.set_button_callback(screen_button_event_handler)
  4798. logger.info("Nextion 屏幕连接成功")
  4799. panels = get_sorted_panels()
  4800. if panels:
  4801. refresh_screen_panel(panels[0][0])
  4802. else:
  4803. # 没有配线架时也要把 IP/版本/MQTT 状态刷到屏幕
  4804. _send_screen_header()
  4805. else:
  4806. logger.warning(f"Nextion 屏幕连接失败: {msg}")
  4807. except Exception as e:
  4808. logger.error(f"Nextion 屏幕初始化异常: {e}")
  4809. # 启动 Nextion 屏幕异步刷新工作线程
  4810. screen_refresh_thread = threading.Thread(target=_screen_refresh_worker, daemon=True)
  4811. screen_refresh_thread.start()
  4812. logger.info("启动 Nextion 屏幕异步刷新线程")
  4813. # 启动网络 IP 监测线程:插网线获取到 IP 后及时写入串口屏
  4814. network_ip_monitor_thread = threading.Thread(target=_network_ip_monitor_loop, daemon=True)
  4815. network_ip_monitor_thread.start()
  4816. logger.info("启动网络 IP 监测线程")
  4817. # 启动服务
  4818. socketio.run(
  4819. app,
  4820. host=FLASK_HOST,
  4821. port=FLASK_PORT,
  4822. debug=FLASK_DEBUG,
  4823. use_reloader=False, # 禁用重载器以避免重复初始化问题
  4824. log_output=False, # 禁用Flask的日志输出,使用我们自己的日志配置
  4825. allow_unsafe_werkzeug=True
  4826. )
  4827. except KeyboardInterrupt:
  4828. # 优雅退出
  4829. logger.info('正在关闭应用...')
  4830. try:
  4831. if isinstance(serial_client.get_status(), dict) and serial_client.get_status().get("connected", False):
  4832. serial_client.disconnect()
  4833. logger.info('串口连接已断开')
  4834. if _mqtt_connected():
  4835. mqtt_client.disconnect()
  4836. logger.info('MQTT连接已断开')
  4837. if dht11_sensor is not None:
  4838. dht11_sensor.stop()
  4839. logger.info('DHT11 传感器线程已停止')
  4840. except Exception as e:
  4841. logger.error(f'关闭连接时出错: {str(e)}')
  4842. # 清理WebSocket连接
  4843. for client_type in connected_clients:
  4844. connected_clients[client_type].clear()
  4845. logger.info('应用已安全关闭')
  4846. except Exception as e:
  4847. logger.exception(f'应用启动失败') # 使用exception记录完整堆栈
  4848. # 确保资源被释放
  4849. try:
  4850. serial_client.disconnect()
  4851. mqtt_client.disconnect()
  4852. except:
  4853. pass