def from_file(cls, filename):
# unzip vsdx file to temp. folder
directory = './{}'.format(filename.rsplit('.', 1)[0])
with zipfile.ZipFile(filename, "r") as zip_ref:
zip_ref.extractall(directory)
# Read [Content_Types].xml
content_types = ContentTypes.from_xml(directory +
'/[Content_Types].xml')
# Read relationships
package_rels = Relationship.from_xml(
'{}/_rels/.rels'.format(directory))
document_rels = Relationship.from_xml(
'{}/visio/_rels/document.xml.rels'.format(directory))
# Read pages and relationships
page_collection = PageCollection.from_xml(directory, content_types)
# Remove extracted folder again
shutil.rmtree(directory)
return cls(page_collection=page_collection,
package_rels=package_rels,
document_rels=document_rels,
content_types=content_types)
def from_xml(cls, dir, content_types):
"""Generate PageCollection from files
:param dir: The directory of the extracted visio package
:param content_types: Instance of :class:`ContentType`
"""
rel_dir = '{}/visio/pages/_rels/'.format(dir)
page_dir = '{}/visio/pages/'.format(dir)
pages = []
rels = Relationship.from_xml(rel_dir + 'pages.xml.rels')
# Parse pages.xml for page info
tree = ET.parse(page_dir + 'pages.xml')
root = tree.getroot()
for child in root:
name = child.attrib.get('Name', '')
id = child.attrib['ID']
# TODO Parse these namespaces properly
rel_id = child[1].attrib[
'{http://schemas.openxmlformats.org/officeDocument/2006/relationships}id']
pages.append(
Page.from_xml(page_dir + rels.rels[rel_id][0], name, id,
rel_id))
return cls(content_types, rels=rels, pages=pages)
def createRelationship(self, charA, charB, relType):
totalRelationships = self.getTotalRelationships()
charATotalTypedRelationships = len(charA.relationships[relType])
charBTotalTypedRelationships = len(charB.relationships[relType])
totalRelationshipsByParticipantsKeys = len(
[x for x in self.relationshipsByParticipants.values() for item in x]
)
# Create relationship object (does not affect state - not binding relationship
# until stored somewhere)
rel = Relationship(self.getTotalRelationships(), charA, charB, relType)
# Don't add relationship if already related
if charB in charA.relationsByType[rel.type]:
print("WARNING: Attempted to create duplicate relationship ",
"({0}) between {1} and {2}".format(rel.type.name, charA.name, charB.name))
return False
# Store relationship object keyed by relationship type
self.relationships[rel.type].append(rel)
# Also store rel obj keyed by participant tuples (in both orderings)
self.storeRelationshipByParticipants(charA, charB, rel)
# Store in characters
charA.addRelationship(charB, rel)
charB.addRelationship(charA, rel)
# Test relationship creation did not have problems
if (totalRelationships+1 != self.getTotalRelationships()):
print("ERROR: Relationship not correctly added to total set")
if (charATotalTypedRelationships+1 != len(charA.relationships[relType])):
print("ERROR: Relationship not correctly added to charA")
if (charBTotalTypedRelationships+1 != len(charB.relationships[relType])):
print("ERROR: Relationship not correctly added to charB")
if (totalRelationshipsByParticipantsKeys+2 != len([x for x in self.relationshipsByParticipants.values() for item in x])):
print("ERROR: Relationship not correctly added, keyed by participants")
return True
def __init__(self, content_types, **kwargs):
"""Initialise pages
:param content_types: Instance of :class:`ContentType`
:param rels: Instance of :class:`Relationship` from pages.xml.rels
:param pages: List of :class:`Page` classes
"""
self.content_types = content_types
self.rels = kwargs.get('rels', Relationship())
self.pages = kwargs.get('pages', [])
@site :
@file: serializers.py
@time: 2017/3/29 下午3:32
@SOFTWARE:PyCharm
"""
from django.contrib.auth.models import Group
from rest_framework import serializers
from app_auth.models import User, Fans, Suggest, InsBinding
import re
from dao.models.music import AllSongMusic
from relationships import Relationship
import redis
from khufu.settings import redis_ship, userpic_url
from social_django.models import UserSocialAuth
r = Relationship(redis_ship)
class FansSerializer(serializers.ModelSerializer):
class Meta:
model = Fans
fields = ('id', 'owner_id', 'follower_id')
# depth = 3
class FollowersSerializer(serializers.ModelSerializer):
"""序列化某人的粉丝"""
is_following = serializers.SerializerMethodField()
picture = serializers.SerializerMethodField()
is_showbtn = serializers.SerializerMethodField()
def relationships(self):
"""Returns a list of document.xml.rels relationship XML tags"""
from relationships import Relationship
return [Relationship(root) for root in self._relationships.getroot().getchildren()]
def __init__(self, **kwargs):
# Content_Types
self.content_types = kwargs.get('content_types', ContentTypes())
# Page collection
self.page_collection = kwargs.get('page_collection',
PageCollection(self.content_types))
# Relationships
# TODO doesn't deserver the schoonheidsprijs here...
self.package_rels = kwargs.get('package_rels', None)
if not self.package_rels:
self.package_rels = Relationship()
self.package_rels.add(
"rId3", "docProps/core.xml",
"http://schemas.openxmlformats.org/package/2006/relationships/metadata/core-properties"
)
self.package_rels.add(
"rId2", "docProps/thumbnail.emf",
"http://schemas.openxmlformats.org/package/2006/relationships/metadata/thumbnail"
)
self.package_rels.add(
"rId1", "visio/document.xml",
"http://schemas.microsoft.com/visio/2010/relationships/document"
)
self.package_rels.add(
"rId5", "docProps/custom.xml",
"http://schemas.openxmlformats.org/officeDocument/2006/relationships/custom-properties"
)
self.package_rels.add(
"rId4", "docProps/app.xml",
"http://schemas.openxmlformats.org/officeDocument/2006/relationships/extended-properties"
)
self.document_rels = kwargs.get('document_rels', None)
if not self.document_rels:
self.document_rels = Relationship()
self.document_rels.add(
"rId2", "windows.xml",
"http://schemas.microsoft.com/visio/2010/relationships/windows"
)
self.document_rels.add(
"rId1", "pages/pages.xml",
"http://schemas.microsoft.com/visio/2010/relationships/pages")
# Document properties
self.doc_props = DocProps()
self.windows_properties = WindowsProperties()
self.document_properties = DocumentProperties()
# apps.xml data
self.company = ''
self.manager = ''
# core.xml data
self.creator = ''
self.language = 'en-US'
self.keywords = []
self.title = ''
self.subject = ''
self.description = ''
# custom.xml data
self.is_metric = True # Using the metric system
class Document:
"""Class holding a visio (*.vsdx) document
The *.vsdx format is a container based on the Open Packaging Conventions
(OPC) introduced by Microsoft. Its a zip file that contains various files
together building a visio document.
This is the tree view of an empty *.vsdx file
.
|- _rels/
| |- .rels
|
|- docProps/
| |- app.xml
| |- core.xml
| |- custom.xml
| |- thumbnail.emf
|
|- visio/
| |- _rels
| | |- document.xml.rels
| |
| |- pages
| | |- _rels
| | | |- pages.xml.rels
| | |
| | |- page1.xml
| | |- pages.xml
| |
| |- document.xml
| |- windows.xml
|
|- [Content_Types].xml
"""
def __init__(self, **kwargs):
# Content_Types
self.content_types = kwargs.get('content_types', ContentTypes())
# Page collection
self.page_collection = kwargs.get('page_collection',
PageCollection(self.content_types))
# Relationships
# TODO doesn't deserver the schoonheidsprijs here...
self.package_rels = kwargs.get('package_rels', None)
if not self.package_rels:
self.package_rels = Relationship()
self.package_rels.add(
"rId3", "docProps/core.xml",
"http://schemas.openxmlformats.org/package/2006/relationships/metadata/core-properties"
)
self.package_rels.add(
"rId2", "docProps/thumbnail.emf",
"http://schemas.openxmlformats.org/package/2006/relationships/metadata/thumbnail"
)
self.package_rels.add(
"rId1", "visio/document.xml",
"http://schemas.microsoft.com/visio/2010/relationships/document"
)
self.package_rels.add(
"rId5", "docProps/custom.xml",
"http://schemas.openxmlformats.org/officeDocument/2006/relationships/custom-properties"
)
self.package_rels.add(
"rId4", "docProps/app.xml",
"http://schemas.openxmlformats.org/officeDocument/2006/relationships/extended-properties"
)
self.document_rels = kwargs.get('document_rels', None)
if not self.document_rels:
self.document_rels = Relationship()
self.document_rels.add(
"rId2", "windows.xml",
"http://schemas.microsoft.com/visio/2010/relationships/windows"
)
self.document_rels.add(
"rId1", "pages/pages.xml",
"http://schemas.microsoft.com/visio/2010/relationships/pages")
# Document properties
self.doc_props = DocProps()
self.windows_properties = WindowsProperties()
self.document_properties = DocumentProperties()
# apps.xml data
self.company = ''
self.manager = ''
# core.xml data
self.creator = ''
self.language = 'en-US'
self.keywords = []
self.title = ''
self.subject = ''
self.description = ''
# custom.xml data
self.is_metric = True # Using the metric system
def to_file(self, filename, tmp_folder='./tmp'):
"""Writes visio diagram to file
:param filename: The filename to write to
"""
if filename.endswith('.vsdx'):
filename.strip('.vsdx')
# Create the directory structure
if not os.path.exists(tmp_folder):
os.makedirs(tmp_folder)
os.makedirs(tmp_folder + '/_rels')
os.makedirs(tmp_folder + '/docProps')
os.makedirs(tmp_folder + '/visio')
os.makedirs(tmp_folder + '/visio/_rels')
os.makedirs(tmp_folder + '/visio/pages')
os.makedirs(tmp_folder + '/visio/pages/_rels')
# TODO the masters folder seems to be generated when you
# create a connection. Lets tackle this when i've got
# the basics worked out.
# os.makedirs(tmp_folder + '/visio/masters')
# os.makedirs(tmp_folder + '/visio/masters/_rels')
# TODO the theme folder lets you put themes on your diagram?
# not present when theres no theme customisation
# os.makedirs(tmp_folder + '/visio/theme')
else:
raise IOError('Error writing to file {}. '
'tmp folder {} already exists'.format(
filename, tmp_folder))
xml_decl = '<?xml version="1.0" encoding="utf-8" ?>'
xml_decl_standalone = '<?xml version="1.0" encoding="utf-8" standalone="yes" ?>'
# Create [content_Types].xml
with open(tmp_folder + '/[Content_Types].xml', 'w') as f:
f.write(xml_decl_standalone)
f.write(self.content_types.to_xml())
# Create docProps files
app_xml, core_xml, custom_xml = self.doc_props.to_xml()
with open('{}/docProps/app.xml'.format(tmp_folder), 'w') as f:
f.write(xml_decl_standalone)
f.write(app_xml)
with open('{}/docProps/core.xml'.format(tmp_folder), 'w') as f:
f.write(xml_decl_standalone)
f.write(core_xml)
with open('{}/docProps/custom.xml'.format(tmp_folder), 'w') as f:
f.write(xml_decl_standalone)
f.write(custom_xml)
shutil.copy('thumbnail.emf', '{}/docProps/'.format(tmp_folder))
# Write pages.xml and pages.xml.rels
pages_xml, pages_xml_rels = self.page_collection.to_xml()
with open('{}/visio/pages/_rels/pages.xml.rels'.format(tmp_folder),
'w') as f:
f.write(xml_decl_standalone)
f.write(pages_xml_rels)
with open('{}/visio/pages/pages.xml'.format(tmp_folder), 'w') as f:
f.write(xml_decl)
f.write(pages_xml)
# Write page?.xml and page?.xml.rels
# TODO, page?.xml.rels not generated yet
for page in self.page_collection.pages:
with open('{}/visio/pages/{}'.format(tmp_folder, page.filename),
'w') as f:
f.write(xml_decl)
f.write(page.to_xml())
# Create _rels files
xml_declaration = '<?xml version="1.0" encoding="UTF-8" standalone="yes"?>'
with open('{}/_rels/.rels'.format(tmp_folder), 'w') as f:
f.write(xml_decl_standalone)
f.write(self.package_rels.to_xml())
with open('{}/visio/_rels/document.xml.rels'.format(tmp_folder),
'w') as f:
f.write(xml_decl_standalone)
f.write(self.document_rels.to_xml())
# Create visio document and window properties
with open(tmp_folder + '/visio/windows.xml', 'w') as f:
f.write(self.windows_properties.to_xml())
with open(tmp_folder + '/visio/document.xml', 'w') as f:
f.write(self.document_properties.to_xml())
# Zip the tmp_folder contents and rename file to vsdx
shutil.make_archive(filename, 'zip', tmp_folder)
shutil.move(filename + '.zip', filename + '.vsdx')
# Remove the temporary folder
shutil.rmtree(tmp_folder)
@classmethod
def from_file(cls, filename):
# unzip vsdx file to temp. folder
directory = './{}'.format(filename.rsplit('.', 1)[0])
with zipfile.ZipFile(filename, "r") as zip_ref:
zip_ref.extractall(directory)
# Read [Content_Types].xml
content_types = ContentTypes.from_xml(directory +
'/[Content_Types].xml')
# Read relationships
package_rels = Relationship.from_xml(
'{}/_rels/.rels'.format(directory))
document_rels = Relationship.from_xml(
'{}/visio/_rels/document.xml.rels'.format(directory))
# Read pages and relationships
page_collection = PageCollection.from_xml(directory, content_types)
# Remove extracted folder again
shutil.rmtree(directory)
return cls(page_collection=page_collection,
package_rels=package_rels,
document_rels=document_rels,
content_types=content_types)
def add_page(self, name):
"""Add a page to the document"""
return self.page_collection.add_page(name)
def add_shape(self, page_rel_id, **kwargs):
return self.page_collection.add_shape(page_rel_id, **kwargs)
def add_connect(self, page_rel_id, shape1, shape2):
return self.page_collection.add_shape(page_rel_id, shape1, shape2)