Как изменить атрибут в Scala XML Element

Question

У меня есть XML-файл, для которого я хотел бы сопоставить некоторые атрибуты с помощью скрипта. Например:

<a>
  <b attr1 = "100" attr2 = "50"/>
</a>

может иметь атрибуты, масштабированные в два раза:

<a>
  <b attr1 = "200" attr2 = "100"/>
</a>

На этой странице есть предложение для добавления атрибутов , но не детализирован способ сопоставления текущего атрибута с функцией (этот способ очень усложнит) http://www.scalaclass.com/book/export/html/1

Что я придумал, так это вручную создать связанный список XML (не scala) ... что-то вроде:

// a typical match case for running thru XML elements:
case  Elem(prefix, e, attributes, scope, children @ _*) => {
 var newAttribs = attributes
 for(attr <- newAttribs)  attr.key match {
  case "attr1" => newAttribs = attribs.append(new UnprefixedAttribute("attr1", (attr.value.head.text.toFloat * 2.0f).toString, attr.next))
  case "attr2" => newAttribs = attribs.append(new UnprefixedAttribute("attr2", (attr.value.head.text.toFloat * 2.0f).toString, attr.next))
  case _ =>
 }
 Elem(prefix, e, newAttribs, scope, updateSubNode(children) : _*)  // set new attribs and process the child elements
}

Это отвратительно, многословно и без необходимости переупорядочивает атрибуты в выводе, что плохо для моего текущего проекта из-за какого-то плохого клиентского кода. Есть ли способ сделать это по-скальски?

Answer 1

Хорошо, лучшее усилие, Scala 2.8. Нам нужно восстановить атрибуты, что означает, что мы должны правильно их разложить. Давайте создадим для этого функцию:

import scala.xml._

case class GenAttr(pre: Option[String], 
                   key: String, 
                   value: Seq[Node], 
                   next: MetaData) {
  def toMetaData = Attribute(pre, key, value, next)
}

def decomposeMetaData(m: MetaData): Option[GenAttr] = m match {
  case Null => None
  case PrefixedAttribute(pre, key, value, next) => 
    Some(GenAttr(Some(pre), key, value, next))
  case UnprefixedAttribute(key, value, next) => 
    Some(GenAttr(None, key, value, next))
}

Далее, давайте разберем связанные атрибуты в последовательность:

def unchainMetaData(m: MetaData): Iterable[GenAttr] = 
  m flatMap (decomposeMetaData)

На данный момент мы можем легко манипулировать этим списком:

def doubleValues(l: Iterable[GenAttr]) = l map {
  case g @ GenAttr(_, _, Text(v), _) if v matches "\\d+" => 
    g.copy(value = Text(v.toInt * 2 toString))
  case other => other
}

Теперь вернемся снова:

def chainMetaData(l: Iterable[GenAttr]): MetaData = l match {
  case Nil => Null
  case head :: tail => head.copy(next = chainMetaData(tail)).toMetaData
}

Теперь нам нужно только создать функцию, которая позаботится об этих вещах:

def mapMetaData(m: MetaData)(f: GenAttr => GenAttr): MetaData = 
  chainMetaData(unchainMetaData(m).map(f))

Так что мы можем использовать это так:

import scala.xml.transform._

val attribs = Set("attr1", "attr2")
val rr = new RewriteRule {
  override def transform(n: Node): Seq[Node] = (n match {
    case e: Elem =>
      e.copy(attributes = mapMetaData(e.attributes) {
        case g @ GenAttr(_, key, Text(v), _) if attribs contains key =>
          g.copy(value = Text(v.toInt * 2 toString))
        case other => other
      })
    case other => other
  }).toSeq
}
val rt = new RuleTransformer(rr)

Что в итоге позволило вам сделать перевод, который вы хотели:

rt.transform(<a><b attr1="100" attr2="50"></b></a>)

Все это можно упростить, если:

• Атрибут фактически определенного префикса, ключа и значения с необязательным префиксом
• Атрибут был последовательностью, а не цепочкой
• Атрибут имел карту, mapKeys, mapValues ​​
• Элем имел карту Атрибут

Answer 2

Вот как вы можете сделать это, используя Scala 2.10:

import scala.xml._
import scala.xml.transform._

val xml1 = <a><b attr1="100" attr2="50"></b></a>

val rule1 = new RewriteRule {
  override def transform(n: Node) = n match {
    case e @ <b>{_*}</b> => e.asInstanceOf[Elem] % 
      Attribute(null, "attr1", "200", 
      Attribute(null, "attr2", "100", Null))
    case _ => n 
  }
}

val xml2 = new RuleTransformer(rule1).transform(xml1)

Answer 3

Так что, если бы я был на вашем месте, я думаю, что я действительно хотел бы написать что-то вроде:

case elem: Elem => elem.copy(attributes=
  for (attr <- elem.attributes) yield attr match {
    case attr@Attribute("attr1", _, _) =>
      attr.copy(value=attr.value.text.toInt * 2)
    case attr@Attribute("attr2", _, _) =>
      attr.copy(value=attr.value.text.toInt * -1)
    case other => other
  }
)

Есть две причины, по которым это не будет работать из коробки:

1. Attribute не имеет полезного метода copy, а
2. Отображение на MetaData дает Iterable[MetaData] вместо MetaData, поэтому даже такая простая вещь, как elem.copy(attributes=elem.attributes.map(x => x)), не будет выполнена.

Чтобы исправить первую проблему, мы будем использовать неявный, чтобы добавить лучший метод копирования в Attribute:

implicit def addGoodCopyToAttribute(attr: Attribute) = new {
  def goodcopy(key: String = attr.key, value: Any = attr.value): Attribute =
    Attribute(attr.pre, key, Text(value.toString), attr.next)
}

Его нельзя назвать copy, поскольку метод с таким именем уже существует, поэтому мы просто вызовем его goodcopy. (Кроме того, если вы когда-либо создаете значения, которые Seq[Node] вместо вещей, которые должны быть преобразованы в строки, вы можете быть немного осторожнее с value, но для наших текущих целей это не обязательно.)

Чтобы решить вторую проблему, мы будем использовать неявное объяснение, как создать MetaData из Iterable[MetaData]:

implicit def iterableToMetaData(items: Iterable[MetaData]): MetaData = {
  items match {
    case Nil => Null
    case head :: tail => head.copy(next=iterableToMetaData(tail))
  }
}

Тогда вы можете написать код, очень похожий на то, что я предлагал в начале:

scala> val elem = <b attr1 = "100" attr2 = "50"/>
elem: scala.xml.Elem = <b attr1="100" attr2="50"></b>

scala> elem.copy(attributes=
     |   for (attr <- elem.attributes) yield attr match {
     |     case attr@Attribute("attr1", _, _) =>
     |       attr.goodcopy(value=attr.value.text.toInt * 2)
     |     case attr@Attribute("attr2", _, _) =>
     |       attr.goodcopy(value=attr.value.text.toInt * -1)
     |     case other => other
     |   }
     | )
res1: scala.xml.Elem = <b attr1="200" attr2="-50"></b>

Answer 4

С помощью Scalate's Scuery и его селекторов и преобразователей CSS3:

def modAttr(name: String, fn: Option[String] => Option[String])(node: Node) = node match {
  case e: Elem =>
    fn(e.attribute(name).map(_.toString))
      .map { newVal => e % Attribute(name, Text(newVal), e.attributes.remove(name)) }
      .getOrElse(e)
}

$("#foo > div[bar]")(modAttr("bar", _ => Some("hello")))

- это преобразование, например это

<div id="foo"><div bar="..."/></div>

в

<div id="foo"><div bar="hello"/></div>`

Answer 5

Мне было проще создать отдельный фрагмент XML и объединить. Этот фрагмент кода также демонстрирует удаление элементов, добавление дополнительных элементов и использование переменных в литерале XML:

val alt = orig.copy(
  child = orig.child.flatMap {
    case b: Elem if b.label == "b" =>
      val attr2Value = "100"
      val x = <x attr1="200" attr2={attr2Value}/>  //////////////////// Snippet
      Some(b.copy(attributes = b.attributes.append(x.attributes)))

    // Will remove any <remove-me some-attrib="specific value"/> elems
    case removeMe: Elem if isElem(removeMe, "remove-me", "some-attrib" -> "specific value") => 
      None

    case keep => Some(keep)
  }
    ++
      <added-elem name="..."/>

// Tests whether the given element has the given label
private def isElem(elem: Elem, desiredLabel: String, attribValue: (String, String)): Boolean = {
  elem.label == desiredLabel && elem.attribute(attribValue._1).exists(_.text == attribValue._2)
}

Для других новичков в Scala XML вам также потребуется добавить отдельный модуль Scala для использования XML в коде Scala.