poll() function is now a static method in python, this is more correct, matching C where the operator is not created to run poll.
def poll(self, context): ...
is now...
@staticmethod
def poll(context): ...
Pythons way of doing static methods is a bit odd but cant be helped :|
This does make subclassing poll functions with COMPAT_ENGINES break, so had to modify quite a few scripts for this.
45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
import bpy
|
|
|
|
def main(context):
|
|
obj = context.active_object
|
|
mesh = obj.data
|
|
|
|
is_editmode = (obj.mode == 'EDIT')
|
|
if is_editmode:
|
|
bpy.ops.object.mode_set(mode='OBJECT', toggle=False)
|
|
|
|
|
|
if not mesh.active_uv_texture:
|
|
bpy.ops.mesh.uv_texture_add()
|
|
|
|
# adjust UVs
|
|
for i, uv in enumerate(mesh.active_uv_texture.data):
|
|
uvs = uv.uv1, uv.uv2, uv.uv3, uv.uv4
|
|
for j, v_idx in enumerate(mesh.faces[i].verts):
|
|
if uv.select_uv[j]:
|
|
# apply the location of the vertex as a UV
|
|
uvs[j][:] = mesh.verts[v_idx].co.xy
|
|
|
|
|
|
if is_editmode:
|
|
bpy.ops.object.mode_set(mode='EDIT', toggle=False)
|
|
|
|
|
|
class UvOperator(bpy.types.Operator):
|
|
'''UV Operator description'''
|
|
bl_idname = "uv.simple_operator"
|
|
bl_label = "Simple UV Operator"
|
|
|
|
@staticmethod
|
|
def poll(context):
|
|
obj = context.active_object
|
|
return (obj and obj.type == 'MESH')
|
|
|
|
def execute(self, context):
|
|
main(context)
|
|
return {'FINISHED'}
|
|
|
|
|
|
if __name__ == "__main__":
|
|
bpy.ops.uv.simple_operator()
|