Как изменить атрибут в элементе Scala XML
У меня есть XML файл, который я хотел бы сопоставить с некоторыми атрибутами in с script. Например:
<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
}
Его отвратительный, многословный и бесполезно переупорядочивает атрибуты в выходе, что плохо для моего текущего проекта из-за плохого кода клиента. Есть ли scala -образный способ сделать это?
Ответы
Ответ 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
- У Elem был атрибут mapAttribute
Ответ 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)
Ответ 3
Итак, если бы я был на вашем месте, я думаю, что я действительно хочу писать, это что-то вроде:
case elem: Elem => elem.copy(attributes=
for (attr <- elem.attributes) yield attr match {
case [email protected]("attr1", _, _) =>
attr.copy(value=attr.value.text.toInt * 2)
case [email protected]("attr2", _, _) =>
attr.copy(value=attr.value.text.toInt * -1)
case other => other
}
)
Есть две причины, по которым это не будет работать:
-
Attribute
не имеет полезного метода copy
и
- Отображение по
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 [email protected]("attr1", _, _) =>
| attr.goodcopy(value=attr.value.text.toInt * 2)
| case [email protected]("attr2", _, _) =>
| attr.goodcopy(value=attr.value.text.toInt * -1)
| case other => other
| }
| )
res1: scala.xml.Elem = <b attr1="200" attr2="-50"></b>
Ответ 4
С помощью Scalate 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>`
Ответ 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.