首頁 > 軟體

python利用lxml庫剩下操作svg圖片

2023-01-22 14:01:01

在大多數場景中,我們都用 lxml 庫解析網頁原始碼,但你是否知道,lxml 庫也是可以操作 svg 圖片的。我們可以使用 lxml 中的 etree 模組來解析 SVG 檔案,然後使用 SVG 中的各種元素和屬性來進行操作。

lxml 操作 svg 圖片範例

在本篇部落格的入門篇,我們首先使用一下 lxml 庫解析 svg 檔案,並修改它的顏色。

SVG 圖片生成,可以檢視 《Python 生成 svg 圖片,一篇部落格帶你掌握 Python 與 svg 之間的操作》 這篇部落格。

借用 svgwrite 庫,生成一個紅色背景的 svg 圖片。

import svgwrite

dwg = svgwrite.Drawing('demo.svg', size=(100, 20), profile='tiny')
dwg.add(dwg.rect(insert=(0, 0), size=(100, 20), fill='red'))
phone_number = '橡皮擦'
dwg.add(dwg.text(phone_number, insert=(0, 15), fill='white', font_size=12))

dwg.save()

同時檢視一下 svg 內容,如下所示。

<?xml version="1.0" encoding="utf-8" ?>
<svg baseProfile="tiny" height="20" version="1.2" width="100" xmlns="http://www.w3.org/2000/svg"
     xmlns:ev="http://www.w3.org/2001/xml-events" xmlns:xlink="http://www.w3.org/1999/xlink">
    <defs/>
    <rect fill="red" height="20" width="100" x="0" y="0"/>
    <text fill="white" font-size="12" x="0" y="15">橡皮擦</text>
</svg>

我們需要替換的內容是 fill="white" ,將其修改為 fill="#03a9f4"。使用 lxml 讀取該檔案,進行顏色替換。

from lxml import etree

# 解析SVG檔案
svg_file = 'demo.svg'
with open(svg_file, 'rb') as f:
    svg_data = f.read()
    parser = etree.XMLParser(remove_blank_text=True)
    svg_tree = etree.fromstring(svg_data, parser)

# 修改顏色
for element in svg_tree.iter():
    if 'fill' in element.attrib:
        # 替換顏色
        element.attrib['fill'] = element.attrib['fill'].replace('white', '#03a9f4')

# 儲存修改後的SVG檔案
with open('modified_demo.svg', 'wb') as f:
    f.write(etree.tostring(svg_tree, pretty_print=True))

此時替換之後,再次檢視 svg 圖片原始碼內容,如下所示。

<svg xmlns="http://www.w3.org/2000/svg" xmlns:ev="http://www.w3.org/2001/xml-events" xmlns:xlink="http://www.w3.org/1999/xlink" baseProfile="tiny" height="20" version="1.2" width="100">
  <defs/>
  <rect fill="red" height="20" width="100" x="0" y="0"/>
  <text fill="#03a9f4" font-size="12" x="0" y="15">&#27233;&#30382;&#25830;</text>
</svg>

重點注意 text 元素部分,發現 fill 屬性的值已經被修改成最新的顏色了。

lxml 給 svg 圖片新增新元素

使用 append() 函數可以給 svg 圖片新增新元素,例如下述程式碼將新增一個矩形到圖片中。

from lxml import etree

# 解析SVG檔案
svg_file = 'demo.svg'
with open(svg_file, 'rb') as f:
    svg_data = f.read()
    parser = etree.XMLParser(remove_blank_text=True)
    svg_tree = etree.fromstring(svg_data, parser)

new_element = etree.Element('rect', x='5', y='5', width='50', height='20', style='fill:#03a9f4')
svg_tree.append(new_element)



# 儲存修改後的SVG檔案
with open('modified_demo.svg', 'wb') as f:
    f.write(etree.tostring(svg_tree, pretty_print=True))

這裡矩形的定位並沒有精確計算,實現的效果圖如下所示。

lxml 刪除 svg 圖片中的元素

除新增外,還可以對 svg 中元素進行刪除操作。範例程式碼如下所示。

from lxml import etree

# 解析SVG檔案
svg_file = 'modified_demo.svg'
with open(svg_file, 'r') as f:
    svg_data = f.read()
    parser = etree.XMLParser(remove_blank_text=True)
    svg_tree = etree.fromstring(svg_data, parser)

# 通過 xpath 查詢需要刪除的元素
elements_to_remove = svg_tree.xpath("//text")
print(elements_to_remove)
for element in elements_to_remove:
    element.getparent().remove(element)

執行程式碼會發現通過 xpath 無法查詢到目標元素 text,這時因為 SVG 檔案中含有名稱空間,導致 xpath 語法查詢不到相應的元素,可以通過為 xpath 語法指定名稱空間來解決這個問題。

新增名稱空間之後的提取語法如下所示。

from lxml import etree

# 解析SVG檔案
svg_file = 'modified_demo.svg'
svg_tree = etree.parse(svg_file)
root = svg_tree.getroot()


# 獲取名稱空間
ns = {'svg': root.nsmap[None]}
# 通過 xpath 查詢需要刪除的元素
elements_to_remove = svg_tree.xpath("//svg:text", namespaces=ns)

print(elements_to_remove)

尤其注意 xpath 部分需要使用 //svg:text 進行提取。

如果不提前生命 ns 變數,可以使用下述程式碼進行指定名稱空間。

elements_to_remove = svg_tree.xpath("//svg:circle", namespaces={'svg': 'http://www.w3.org/2000/svg'})

有了上述基礎之後,完整的 lxml 刪除元素程式碼如下所示。

from lxml import etree

# 解析SVG檔案
svg_file = 'modified_demo.svg'
svg_tree = etree.parse(svg_file)
root = svg_tree.getroot()


# 獲取名稱空間
ns = {'svg': root.nsmap[None]}
# 通過 xpath 查詢需要刪除的元素
elements_to_remove = svg_tree.xpath("//svg:text", namespaces=ns)

print(elements_to_remove)
for element in elements_to_remove:
    element.getparent().remove(element)

# 儲存修改後的 SVG 檔案
with open('modified_demo1.svg', 'wb') as f:
    f.write(etree.tostring(svg_tree, pretty_print=True))

以上就是python利用lxml庫剩下操作svg圖片的詳細內容,更多關於python lxml操作svg的資料請關注it145.com其它相關文章!


IT145.com E-mail:sddin#qq.com