o hH @sdZddlmZddlmZddlmZmZddlm Z m Z m Z m Z dZ ddZd d Zd d ZGd dde ZddZedkrEedSdS)a?Invokes the Java tsurgeon on a list of trees Included with CoreNLP is a mechanism for modifying trees based on existing patterns within a tree. The patterns are found using tregex: https://nlp.stanford.edu/nlp/javadoc/javanlp/edu/stanford/nlp/trees/tregex/TregexPattern.html The modifications are then performed using tsurgeon: https://nlp.stanford.edu/nlp/javadoc/javanlp/edu/stanford/nlp/trees/tregex/tsurgeon/Tsurgeon.html This module accepts Tree objects as produced by the conparser and returns the modified trees that result from one or more tsurgeon operations. ) tree_reader)Tree)TsurgeonRequestTsurgeonResponse) send_request build_tree from_treeJavaProtobufContextz=edu.stanford.nlp.trees.tregex.tsurgeon.ProcessTsurgeonRequestcCs t|ttSN)rr TSURGEON_JAVA)requestr Q/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/server/tsurgeon.pysend_tsurgeon_requests rcCst|tr|f}t}|D] }|jt|dq tdd|Dr%|f}|D]%}t|dkr3td|j }|d|_ |ddD]}|j |qCq'|S)z Build the TsurgeonRequest object trees: a list of trees operations: a list of (tregex, tsurgeon, tsurgeon, ...) gcss|]}t|tVqdSr ) isinstancestr).0xr r r )sz build_request..z6Expected [tregex, tsurgeon, ...] but just got a tregexrN) rrrtreesappendralllen ValueError operationsaddtregextsurgeon)rrr tree operationoperation_requestrr r r build_requests    r"cGs"t||}t|}dd|jDS)z Returns the result of processing the given tsurgeon operations on the given trees Returns a list of modified trees, eg, the result is already processed cSg|]}t|dqSrrrtr r r >z!process_trees..)r"rr)rrr resultr r r process_trees5s r+cs*eZdZdZdfdd ZddZZS)Tsurgeonz Tsurgeon context window This is a context window which keeps a process open. Should allow for multiple requests without launching new java processes each time. Ncstt||ttdSr )superr,__init__rr )self classpath __class__r rr.HszTsurgeon.__init__cGs$t||}||}dd|jDS)NcSr#r$r%r&r r rr(Nr)z$Tsurgeon.process..)r"process_requestr)r/rrr r*r r rprocessKs  zTsurgeon.processr )__name__ __module__ __qualname____doc__r.r4 __classcell__r r r1rr,Asr,cCs0d}t|}d}d}t|||f}t|dS)z7 A small demonstration of a tsurgeon operation z[( (SBARQ (WHNP (WP Who)) (SQ (VP (VBZ sits) (PP (IN in) (NP (DT this) (NN seat))))) (. ?)))zWP=wpzrelabel wp WWWPPPN)r read_treesr+print)textrrrr*r r rmainQs   r=__main__N)r8stanza.models.constituencyr%stanza.models.constituency.parse_treerstanza.protobufrr$stanza.server.java_protobuf_requestsrrrr r rr"r+r,r=r5r r r rs