2424urlreq = util .urlreq
2525
2626GLOBAL_REV_TYPE = "GLOBAL_REV"
27+ CLIENT_CALLER = "sapling"
2728
2829
2930class ClientError (Exception ):
@@ -214,8 +215,12 @@ def _normalizerevisionnumbers(self, *revision_numbers):
214215
215216 def getdiffversion (self , timeout , diffid , version = None ):
216217 query = """
217- query DiffLastVersionDescriptionQuery($diffid: String!){
218+ query DiffLastVersionDescriptionQuery(
219+ $diffid: String!
220+ $client_caller: String!
221+ ) {
218222 phabricator_diff_query(query_params: {
223+ client_caller: $client_caller
219224 numbers: [$diffid]
220225 }) {
221226 results {
@@ -275,7 +280,7 @@ def getdiffversion(self, timeout, diffid, version=None):
275280 extra_query = ""
276281 query = query % extra_query
277282
278- params = {"diffid" : diffid }
283+ params = {"diffid" : diffid , "client_caller" : CLIENT_CALLER }
279284 ret = self ._query (timeout , query , params )
280285
281286 try :
@@ -332,8 +337,12 @@ def getnodes(self, repo, diffids, diff_status, timeout=10):
332337 ret = self ._mocked_responses .pop ()
333338 else :
334339 query = """
335- query DiffToCommitQuery($diffids: [String!]!){
340+ query DiffToCommitQuery(
341+ $diffids: [String!]!
342+ $client_caller: String!
343+ ) {
336344 phabricator_diff_query(query_params: {
345+ client_caller: $client_caller
337346 numbers: $diffids
338347 }) {
339348 results {
@@ -359,7 +368,7 @@ def getnodes(self, repo, diffids, diff_status, timeout=10):
359368 }
360369 }
361370 """
362- params = {"diffids" : diffids }
371+ params = {"diffids" : diffids , "client_caller" : CLIENT_CALLER }
363372 ret = self ._query (timeout , query , params )
364373 # Example result:
365374 # { "data": {
@@ -479,7 +488,12 @@ def getrevisioninfo(self, timeout, signalstatus, *revision_numbers):
479488 if self ._mock :
480489 ret = self ._mocked_responses .pop ()
481490 else :
482- params = {"params" : {"numbers" : rev_numbers }}
491+ params = {
492+ "params" : {
493+ "client_caller" : CLIENT_CALLER ,
494+ "numbers" : rev_numbers ,
495+ }
496+ }
483497 ret = self ._query (timeout , self ._getquery (signalstatus ), params )
484498 return self ._processrevisioninfo (ret )
485499
0 commit comments