如何在小程序或者app中，通过js的手段，在根元素下设置自定义属性值

在h5和pc浏览器，可以在html元素下通过自定义属性可以实现主题定制切换，window.document.documentElement.setAttribute('data-theme', uni.getStorageSync('theme') || 'light')，那么现在要在小程序或者app端，同样实现此效果，那么如何写相对应的代码呢？