成为VIP
前言
最近遇到了一个需求，要求在python的selenium自动化打开的浏览器中，执行js代码，实现获取图片的base64，并返回给python变量。
看似很简单的一个需求，实际上包含了很多js相关的知识点和开发上的技术点。
今天这篇文章，主要写写整个过程中有用的知识点。
原始方法
其实最开始我没有考虑用js获取图片，所以用pyautogui键盘操作下载图片，代码如下：
1import pyautogui
2from selenium.webdriver import ActionChains
3
4driver.get(url)
5ActionChains(driver).move_to_element(driver.find_element_by_tag_name('img')).context_click().perform()
6pyautogui.typewrite(['v'])
7time.sleep(1)
8pyautogui.typewrite(['enter'])
这里面，主要运用电脑上的快捷键，在选择图片后，右键，然后按v键，就会弹出保存窗口，再按enter，就可以按照默认信息保存。
这种方式是模拟手动，原理简单，但是，在多次重复这个操作时，会发现偶尔会出现一个bug，就是按v键会与保存冲突，使得电脑以为将图片名命名为v，然后整个过程就无法继续进行了。
为了解决上述问题，我才采用js的方法，才有了后面的内容。
第一版js代码
1var source = document.getElementsByTagName('img')[0];
2// 在网页元素中找到图片资源
3var img=new Image();
4// 创建一个image元素
5img.src=source.src;
6// 设置img的图片链接为网页图片链接
7img.crossOrigin='anonymous';
8// 设置跨域的配置，这个是比较常见的，如果不了解，读者可以自行百度
9var xhr=new XMLHttpRequest();
10// 初始化一个http请求
11xhr.open('GET',source.src,false);
12// 构造请求
13xhr.send(null);
14// 发送请求
15xhr.open('GET',source.src,false); // 重复上面步骤，为了确定加载到资源
16xhr.send(null);
17var canvas = document.createElement('canvas');
18// 创建canvas元素
19var ctx=canvas.getContext('2d');
20// 获取canvas二维属性
21ctx.drawImage(img,0,0);
22// 将img画到canvas上
23var dataURL = canvas.toDataURL('image/png');
24// 利用canvas转换图片为base64形式
25return dataURL;
上面的代码运行后，发现获取到的base64还原出的图片是黑色的，运行很多次都这样。所以我又在浏览器console下一条一条执行，结果发现能够给出正常的base64，但是在浏览器直接解析后，发现图片虽然能够正常显示，但是图片的大小是被裁剪的。
仔细思考后，我感觉可能是因为获取的图片的问题，但是看了network的加载记录，发现好像并不是这样，加载的图片都是完整的。所以，可能是canvas画图的时候出了一些问题，所以就去了解canvas的drawImage函数。
研究一番，发现，这个函数第一个参数是图片信息js获取url参数，第二和三个参数是画布的坐标，表示从哪个位置开始画，然后还可以有第四和五个参数，表示要画出的图的宽和高，所以我就修改了那句为
1ctx.drawImage(img,0,0,source.naturalWidth,source.naturalHeight);
2// 这里面的source是指网页的图片，然后naturalWidth是原宽，naturalHeight是原高
这里就把整个图画上去，但是经过实验后，我发现画出来的图还是裁剪后的图，而且，每次的图的大小都是一样，这就使我明白，会不会是画布的大小出了问题。
经过查阅资料，我发现canvas是有初始大小的，而且如果不改的话，是不会变的。所以，我又在画图前设置canvas的画布大小。
1canvas.width=source.naturalWidth;
2canvas.height=source.naturalHeight;
经过这次，我在浏览器端一条一条执行时，发现代码执行没问题，最后结果都正确。我以为第一天的工作结束了，解决了这个问题，明天简单弄弄就好了，但是一个更麻烦的问题在路上了。
注意：上面的在console下一条一条执行代码和在python中执行js代码的结果是不一样的。因为一条一条执行，中间是有时间消耗的，不好异步任务可以在此期间完成。
第二版js代码
1var source = document.getElementsByTagName('img')[0];
2// 在网页元素中找到图片资源
3var img=new Image();
4// 创建一个image元素
5img.src=source.src;
6// 设置img的图片链接为网页图片链接
7img.crossOrigin='anonymous';
8// 设置跨域的配置，这个是比较常见的，如果不了解，读者可以自行百度
9var xhr=new XMLHttpRequest();
10// 初始化一个http请求
11xhr.open('GET',source.src,false);
12// 构造请求
13xhr.send(null);
14// 发送请求
15xhr.open('GET',source.src,false); // 重复上面步骤，为了确定加载到资源
16xhr.send(null);
17var canvas = document.createElement('canvas');
18// 创建canvas元素
19canvas.width=source.naturalWidth;
20// 设置画布宽
21canvas.height=source.naturalHeight;
22// 设置画布高
23var ctx=canvas.getContext('2d');
24// 获取canvas二维属性
25ctx.drawImage(img,0,0,source.naturalWidth,source.naturalHeight);
26// 将img画到canvas上
27var dataURL = canvas.toDataURL('image/png');
28// 利用canvas转换图片为base64形式
29return dataURL;
上面代码是前面修改后、我以为正确的代码，但是，第二天放到python中运行后，发现结果还是不对，图片仍然是黑色的，不正确。
我说，难道是我保存的昨天的代码不对吗？所以还是重复错误的步骤，在浏览器下一条一条的运行。结果是正确的，所以我就很纳闷，一样的代码怎么会有不同的结果？
我不理解，但是我感觉可以试试将代码包装从函数的形式，在浏览器端运行。观察一下结果，发现结果给出的内容确实不是图片正常的base64，只能转为黑色的图片。
为了验证我的结果，我还运行了多次，结果都一样。我还将浏览器的正常结果、错误结果和python端的结果进行对比，我发现正确结果是很长的，而两个错误结果一样长度，都很短。
故事到这，陷入了僵局，我不知道该怎么办了。
正好到晚饭点了，所以就先去吃饭吧，吃饭的过程中，看看canvas的文档，了解一下具体的使用方法，说不定会有一些意外的收获。
在晚饭期间，我看到canvas画图的一些注意事项：若调用 drawImage 时，图片没装载完，那什么都不会发生（在一些旧的浏览器中可能会抛出异常）。因此你应该用 load 事件来保证不会在加载完毕之前使用这个图片。（来源于）
官方文档给出了案例：
1var img = new Image(); // 创建 img 元素
2img.src = 'myImage.png'; // 设置图片源地址
3img.onload = function(){
4 // 执行 drawImage 语句
5}
我没有直接将这个作为解决方案，我又多看了一些论坛，发现大部分都是这样实现的。
后面，将这个代码加到我的代码中，我发现return语句在外面的话，还是会返回错值；return语句里面的话，是可以正常执行的，但是返回的值是无法返回到外层的。
于是，我灵机一动，直接return img.onload….
在浏览器下确实成功了，但是，在python这边，却还不对，返回的是一个函数对象。
到此，我明白，要等图片装载完成后，再执行drawimage语句，然后再返回才行。但是，js代码不会等，只要执行了就会继续往下执行下一条，不管你是否执行完成。
第三版js代码
为了解决加载完成的问题，我在想是不是有可以休眠等待的函数，查阅之后，发现可以自己构造一个sleep函数。
1const sleep = (delay) => new Promise((resolve) => setTimeout(resolve, delay));
2sleep(1000); // 休眠1s
但是，加在上面代码后面，返回结果还是那样，感觉无法停止代码执行。
经过查阅资料，发现还有一个setTimeout函数，可以设置一段时间后再执行函数，但是输入值只能是函数，这就要求在函数里面返回最后的base64。但是，实际实验了一下，发现这个方式行不通，不返回内容。
然后，继续查阅资料，发现可以通过throw error抛出内容，我当然也尝试了，发现都是uncaught。我当时就纳闷，然后把setTimeout函数也换成了img.onload，但是结果都不对。
中间经过大量的实验，省略n次尝试。。。
最后，我明白了只有在try里面直接执行throw error代码，才能够catch到。至于利用try包裹etTimeout和img.load等耗时异步语句，都不能正常catch，因为正常执行到throw error时，已经结束了外面了try{}catch{}语句的执行，所以根本catch不到。
上面的一大堆过程，让我明白了，根本原因是在异步代码的执行完成前能否不继续执行下面的代码。
最终的js代码
1var source = document.getElementsByTagName('img')[0];
2var img=new Image();
3img.src=source.src;
4img.crossOrigin='anonymous';
5var canvas = document.createElement('canvas');
6canvas.width=source.naturalWidth;
7canvas.height=source.naturalHeight;
8var ctx=canvas.getContext('2d');
9function httpPromise(){
10 return new Promise((resolve,reject) => {
11 if(img.complete || img.naturalHeight!==0)
12 {
13 ctx.drawImage(img,0,0,source.naturalWidth,source.naturalHeight);
14 var dataURL = canvas.toDataURL('image/png');
15 resolve(dataURL);
16 }
17 else{
18 img.onload=function(){
19 ctx.drawImage(img,0,0,source.naturalWidth,source.naturalHeight);
20 var dataURL = canvas.toDataURL('image/png');
21 resolve(dataURL);
22 }
23 }
24 })
25}
26return httpPromise().then((dataURL)=>{return dataURL;}
最后这版代码做了一些改动，首先我删除了xhr加载的代码，因为这个加载过程和最后的结果没有关系，不影响最后的结果；然后，我添加了Promise异步处理机制，运用这个，就可以等待异步执行结束后，收集所有的结果，并且返回（我记得java也有类似的库可以解决异步等待的问题）；最后，我添加了判断机制，防止img.onload因为缓存的原因不响应，所以分成了两种情况进行处理，但都是将结果收集，在return语句处返回。
插一嘴：这里需要return语句js获取url参数，是因为python里面的driver.excute_script函数是将我们的js代码包装成函数，需要返回内容进行处理，否则返回值就直接为None。
上面就是我通过自己摸索，实现的这样一个需求，虽然经过了很多磕磕绊绊，但是结果还是好的。其实，有一篇博客的代码我觉得是优于我的代码。
大佬的代码
1// https://blog.csdn.net/yaxuan88521/article/details/122794055
2let c = document.createElement('canvas');
3let ctx = c.getContext('2d');
4let img = document.getElementsByTagName('img')[0];
5c.height=img.naturalHeight;
6c.width=img.naturalWidth;
7ctx.drawImage(img, 0, 0,img.naturalWidth, img.naturalHeight);
8let base64String = c.toDataURL();
9return base64String;
上面大佬的代码，是直接获取加载后的图片元素，效果是优于我的代码，毕竟减少了一次加载的时间消耗。
总结
本文通过笔者在js获取图片base64的过程中，遇到的问题。
总的说来，可以分为以下几点：
canvas绘图需要修改画布大小，否则无法画完整。
canvas的drawimage函数，需要等待图片加载完成，否则就会直接返回画布。
js代码本身是同步执行的，但是如果实现了一些异步函数，如setTimeout和img.onload等，可能会出现返回结果不正确的情况，所以，如果需要返回值，最好利用funture构造，最后接收返回值。
对于同一个需求，最好选择更为方便的方法实现，否则会浪费时间和资源。（笔者在这个地方花费时间有点多，但是对于js也有了更深的认识）
最后，本文主要介绍了js获取图片base64的问题，中间过程有些曲折，内容可能有点凌乱，但是技术点还是给出了，希望各位读者能够从中有所收获。
限时特惠：本站每日持续更新海量设计资源，一年会员只需29.9元，全站资源免费下载
站长微信：ziyuanshu688
htebazile
