查看原文
其他

搜罗全网!ArcGIS二次开发Python(arcpy)指南(七):你也可以制作ArcGIS工具箱

你个海怪 GIS荟 2022-05-20

点击上方蓝字,关注我带你飞!

前言:让自定义工具箱变得简单如喝水一样。

 

1.自定义工具箱是什么

就像名字一样,自定义工具箱是由个人创建的工具箱,就像下面这样,可以理解成一个 .tbx 后缀的压缩包,每一个工具箱中都可以添加多个工具集、脚本工具甚至模型构建器工具。

 

一大堆工具箱

 

1.1如何添加自定义工具箱(熟悉可跳过)

第一步:右键点击 ArcToolbox -> 添加工具箱

 

 

第二步:在打开的添加工具箱窗口中选中需要添加的工具,然后点击右下角的打开即可。

 

添加工具箱到 Arcmap

 

1.2如何创建自定义工具箱

点击右上角的红色工具箱符号,可以新建并重命名一个工具箱。

我们这里新创建了一个海怪工具箱,用于之后的教程演示。

 

新建工具箱

 

1.3添加一个脚本工具

新创建的工具箱就像一个大箱子,箱子呢是用来装工具的,你可以直接在工具箱中右键,然后点击添加,再点击脚本,进而创建一个脚本工具;

 

 

亦或是右键,点击新建,再点击工具集,创建一个可以自定义名称的工具集,然后在工具集里新建一个脚本工具。

 

 

这里添加的脚本工具只是一个空壳,我们后面会继续完善。

新建的工具箱位于 ../Chapter7/海怪工具箱.tbx

 

 

2.工具箱的封装

工具箱由三部分组成,其中有两部分需要我们来控制:

  • 一是输入界面:确定工具箱输入界面的参数定义和设置;

  • 二是 Python 源代码:功能的实现,为 .py 后缀的脚本文件;

  • 三是对参数行为提供额外控制的可选验证代码,其位于工具箱中(这部分自带,基本上不用修改)。

     

所以下面的重点就是输入界面和 Python 源代码。

 

2.1工具箱输入界面

工具箱输入界面,或者说图形交互界面(GUI),就像下面的几张图一样,从该界面我们可以控制工具的行为,自定义输入界面然后配合 Python 脚本,可以实现多种客制化的功能。

简单来说,你可以认为输入界面是依托于 Arcmap 的“前端”,脚本代码就是负责运行的“后端”。

 

导入导出工具:

为方便演示,制作一个非常简单的工具箱,它的功能呢就是把输入的要素类(数据库或者 shp )再导出来而已,是完全没有用的导入导出功能啦,不过作为演示还是非常不错的。

 

制作完成后的导入导出工具的界面

 

第一步:

在工具箱中,右键点击添加,再点击脚本,然后修改名称,修改成你想要的名字,勾选存储相对路径名,然后点击下一页。

 

修改名称

 

第二步:

然后添加脚本文件,添加该脚本文件 ../Chapter7/toolscript/input_and_output.py,读者可以在最后的下载包中找到该文件。

然后点击下一页。

 

链接python脚本

 

第三步:

控制用户交互界面。ArcGIS 在这里内置了一整套简单但是实用的图形控件,比如输入菜单、下拉框、多选框、单选等,借由这些图形控件我们可以控制脚本工具的图形交互界面。

我们这里使用的脚本功能非常简单,上面也有说到:输入一个矢量要素类,不做任何处理,然后输出一个矢量要素类。

为了匹配这个脚本,我们需要设置一个输入图层的控件,同时为了能顺利的输出,也需要一个输出图层的控件,注意一个是输入一个是输出。

 

从下图可以看到,左边(左边的橙色框)是显示名称,其中输入输出控件的名称以及输入控件的名称,这里命名为输入要素和输出要素;然后在右边(右边的橙色框)的数据类型下拉框中,我们选择要素图层,这里只能做选择,不能任意输入。

 

设置名称和数据类型

 

那么输入输出是如何区分的呢?

选中输入要素(变成蓝色),然后可以看到下方(下方橙色框)的参数属性设置选项卡,这里可以对参数进行调整。

 

参数选项

 

类型:有三种,一般只会用到 Required 和 Optional,表示必须和可选,用于控制参数是否可以忽略不填;

方向:两种,Input 和 Output,表示输入和输出,大部分情况都是使用 Input 来获得参数,Output 用于获得最后的成果数据,像是矢量、栅格或者图表;

多值:Yes 或者 No,用于输入多个相同类型的数据,工具打开后就是下面这样,在输入要素中可输入多个要素类;

 

 

默认:设置一个控件的默认值,直接输入即可。可配合多值一同使用,使用 ; (英文的分号)分开,

 

   

结果如下,可以看到一打开该工具就已经填充上了预设的多个默认值(报错是因为数据类型不是要素图层);

 

 

环境:工具箱的环境设置,比如掩膜、容差范围、默认数据库位置等,一般不预先设置;

过滤器:选择文件类型可以在选择文件时只显示特定格式的文件;

获取自:指定数据的获取源。这是个什么意思呢?比如我们想要获取图层A的某个字段,获取源就是图层A。

以下面这种图作为例子,新增名为字段的控件(上方的橙色框),数据类型也是字段,然后参数属性中获取自选择输入要素(下方橙色框)。

 

设置字段控件

 

点击完成后,打开脚本工具。在输入要素框中选择一个图层,然后再点击获取字段,如下就可以选择图层中的字段了。

 

获取图层的字段

 

符号系统:在最下面还有一个符号系统,通常只能作用于输出的矢量图层(方向:Output),可为输出图层指定设置好的 lyr 文件样式。

 

设置符号系统

 

第四步:

点击右下方的完成,那么脚本工具的输入界面就完成了。

 

2.2 Python 脚本

工具箱输入界面和 Python 代码要相互配合才行,工具箱界面中的输入、输出以及各种各样的参数,在 Python 脚本文件中都得一一对应上,将输入界面中的各个参数传给 Python 脚本,这样才能让 Python 脚本按照预期的效果正常运行。

 

看上去有点麻烦,但实际上 arcpy 中提供了现成的函数就可以直接获得参数。

 

①参数对接(获取)

至于如何对接参数,我们使用刚刚的那个脚本文件举例:

../Chapter7/toolscript/input_and_output.py

 



# -*- coding:utf-8 -*-import arcpy#▶1◀input_lyr = arcpy.GetParameterAsText(0)output = arcpy.GetParameterAsText(1)if __name__ == '__main__':    lyr = arcpy.mapping.Layer(input_lyr)    arcpy.CopyFeatures_management(lyr, output)

 

▶1◀:arcpy 提供了 GetParameterAsText(index) 函数,该函数根据输入界面的排序号码获得指定的参数。比如  GetParameterAsText(0) 获得的就是输入图层参数, GetParameterAsText(1)  获得的就是第二个参数,也就是输出图层参数。

如果有更多参数,以此类推。

 

②参数对接中应该注意的情况

GetParameterAsText(index) 从输入界面获得的参数都会转变成字符串,没错都是字符串。

如果你只是输入输出地址这种,那么没有问题,因为地址本来就是字符串,但是如果你设置了一个选择长度范围的长整型的控件、亦或是某种布尔值选择,最后输入到 python 脚本这里都会变成字符串。

 

  • 整数变成字符串,使用 int(str) 规避;

  • 布尔型变成字符串,True 变成了 "True",False 变成了 "False",这样就不能判断真假了,需要写一个判断语句或者随便你,方法很多;

  • 还有就是一个输入控件多值的情况,所有值根据先后连接成一个字符串,中间由 、(英文分号)间隔,可以使用 split 函数分开;

  • ...更多的情况都是大同小异,多试几次就会了。

 

③更好的参数获取写法

上面例子是一个一个的获取参数,但是如果参数很多呢?10个参数难道写10遍 GetParameterAsText(index)

 


# -*- coding:utf-8 -*-def in_and_out(in_lyr, out_lyr):    lyr = arcpy.mapping.Layer(in_lyr)    arcpy.CopyFeatures_management(lyr, out_lyr)if __name__ == '__main__':    #▶2◀    argv = tuple(arcpy.GetParameterAsText(i)                 for i in range(arcpy.GetArgumentCount()))    in_and_out(*argv)

 

▶2◀:使用 GetArgumentCount() 函数获取一个有几个参数,然后一次性把所有参数拿到,最后在函数中使用 *argv 解包即可。

 

使用这种方法可以显得比较厉害,同时精简代码。

 

 

④信息输出

难免想要在代码中打 print,不管是作为临时的“断点”检查一下,还是作为长期的信息输出。

那么在工具箱中如何向外面输出信息呢?

arcpy 提供了 AddMessage(str) 方法向外界传递消息。

 


# -*- coding:utf-8 -*-def in_and_out(in_lyr, out_lyr):    lyr = arcpy.mapping.Layer(in_lyr)    #▶3◀        arcpy.AddMessage("Hello World")    arcpy.CopyFeatures_management(lyr, out_lyr)if __name__ == '__main__':        argv = tuple(arcpy.GetParameterAsText(i)                 for i in range(arcpy.GetArgumentCount()))    in_and_out(*argv)

 

▶3◀:我们在这里使用  AddMessage("Hello World"), 然后运行脚本工具,在完成界面可以看到输出的 Hello World 信息。

 

 

 

⑤导入脚本

到这里基本上工具就完成了,只不过 Python 脚本和自定义工具箱是分开,创建的工具在运行的时候会根据相对路径位置(如果你一开始勾选了存储相对路径名)寻找 Python 脚本,找到并运行。

不过你愿意的话,也可以把 Python 脚本文件导入到工具箱中,这样带着一个工具箱就行了。

 

 

 

3. 工具箱的各种错误和注意事项

3.1“限定”的错误

让我如此苦恼、如此害怕的,就是把写好的 Python 脚本封装成 ArcGIS 的工具箱。

 

ArcGIS 工具箱

 

每次我把程序代码写好,也能正常的运行,然后将其导入自定义工具箱中,点击运行之前还得沐浴更衣好好的祈祷一番,愿天公助我。

然而事与愿违,刚刚还好好的程序报错了,错误可谓是千奇百怪,大部分时候都在说谜语,明明显示第100行有什么什么错,结果一看,100行是空行...

显然,ArcGIS 工具箱有一些特别的魔法,拥有一些专属自己的“限定”错误,这种错误只有在这里才能品味到。

还不快跟着我尝一尝新鲜的魔法shit。

 

Note:以下所有错误均是将脚本封装成工具箱后,运行才可能会出现的错误,请注意。
Note:我一直使用的10.3版本,以下的所有错误不对非10.3的版本负责,当然其他版本也差不多。

 

3.2工具箱名字导致的错误

我们先从最简单的错误开始。

工具箱名称导致的错误:

 

 

由于输出路径会自动填入工具箱的名字,中间有特殊符号的话,比如:“-”,就会报错。

 

 

3.2缩进错误

诡异的错误来了,报错显示如下。我都不知道是说的什么鬼话,怎么还有缩进错误?

 

 

锁定到报错提示的地方,是这样的:

 

 

看上去非常正常,100%正常,开始我不信是这里的问题,我开始折腾,狂写 print 语句,使劲打断点,使劲测试,最后机缘巧合之下,我把中文注释移到了上方,然后就顺利运行了。

 


# ...# 存在该字段else:    # ...

 

好吧,由于把 else 和中文注释写到一行,导致 else 被莫名其妙被跳过了,所以导致错误。

 

3.3两种“变量”错误

第一种:

global name xx is not defined

 

第二种:

name xx is not defined

 

 

这个错是缠绕我最久的,每次出这种错误我都到处测试,偶尔莫名其妙就好了,但是一直不知道原因,直到后来我怀疑是不是也是中文字符造成的错误,结果发现,嘿,还真是!

比如错误处的代码是这样的,在变量 pta2 的上面是一行中文注释:

 


# ...# 二象限pta2 = (oX - length * 2, pta[1])# ...

 

我们使用空行将报错的变量和中文注释隔开:

 


# ...# 二象限pta2 = (oX - length * 2, pta[1])# ...

 

于是乎,程序顺利运行了,原来如果在变量的上一行是中文注释,有可能会导致下面那一行被工具箱里面的神奇魔法规则跳过。

自此,困扰我很久、经常让我折腾到深夜的各种暗坑终于有了解决方法。

 

个人的经验,希望能对你有所帮助和启发。

 

 

4.最后和总结


总结:

  • 工具箱只是一个箱子,其中需要你自己创建脚本工具来实现各种功能;

  • 脚本工具和 Python 代码(脚本)两位一体的,互相配合,脚本工具负责图形界面,Python 脚本专注于功能的实现,最后在工具箱中将两者结合起来;

  • 脚本工具添加的图形界面控件也有多种属性,最重要的有控制输入输出的方向属性、是否必须的类型属性,以及其他的偶尔使用的多值默认等属性;

  • GetParameterAsText(index) 方法可从图形输入界面获取参数,不过需要注意的是,返回的结果全部都是字符串,记得转换如果有必要;

  • 脚本封装进工具箱报错多半是由中文引起的,谨记,解决方法参考上文的第三大点。

 

使用版本:

  • Windows 10

  • PyCharm 2021.2.3

  • ArcGIS 10.3

  • Python 2.7.8

 

源代码、教学文档离线小册子下载:

公众号后台回复:at2021 可获得下载链接。

可免费下载整套教程源码、文件,还有PDF文档方便电脑端查看。

欢迎下载,互相学习共同进步!

 


很难的教程、有趣的文章、好看的地图,这里都有!

原创不易,老板点点下方的 收藏在看

 

本系列专辑:

《ArcPy教程指南2021》——从0开始,从实际应用出发,带大家熟悉掌握 ArcPy。(持续更新中!)

 


您可能也对以下帖子感兴趣

文章有问题?点此查看未经处理的缓存