用AI写代码 -- Github Copilot测试

截图为copilot官网

要使用copilot，首先要安装vscode。下载copilot插件。下载完了以后，并不能马上使用。还需要申请内测账号，前几天，我终于收到内测账号通过的邮件。趁着今天在家，连忙打开vscode测试一下。

爬图片

我们的第一个实验是爬csdn的logo。

于是，我新建了一个文件，我开始写注释。当我写道retri的时候，copilot已经知道我要说什么了，直接把一句话补全了。牛！

我于是打了一下tab，这句话就补全了。

接着，我打了import，copilot已经开始建议我导入urllib了。牛！

打完def三个字，整个函数的建议都有了。牛！

接着，打if：

于是，满怀期待的，运行了程序，结果，悲剧了：

AttributeError: module ‘urllib’ has no attribute ‘urlopen’

我还是比较熟悉requests模块，于是，我把第一句代码改成了import requests，重新生成代码。如下：

# find logo in csdn.net
# retrieve the logo and save it in the same directory

import requests
from bs4 import BeautifulSoup
import os

def get_logo(url):
    r = requests.get(url)
    soup = BeautifulSoup(r.text, 'html.parser')
    logo = soup.find('link', rel='shortcut icon')
    if logo:
        logo_url = logo['href']
        r = requests.get(logo_url)
        if r.status_code == 200:
            with open(os.path.basename(logo_url), 'wb') as f:
                f.write(r.content)
                print('logo saved')
        else:
            print('logo not found')
    else:
        print('logo not found')
        

if __name__ == "__main__":
    get_logo('http://www.csdn.net')

运行完了，发现不是logo，是favicon。😦

算了，算了，Copilot找不到logo，还是我自己找吧。

于是，我把注释改成了

# download and save https://img-home.csdnimg.cn/images/20201124032511.png
# to the same folder

生成的代码如下：

# download and save https://img-home.csdnimg.cn/images/20201124032511.png
# to the same folder

import requests
from bs4 import BeautifulSoup
import os

def download():
    url = 'https://img-home.csdnimg.cn/images/20201124032511.png'
    root = './img'
    path = root + url.split('/')[-1]
    try:
        if not os.path.exists(root):
            os.mkdir(root)
        if not os.path.exists(path)