mschoi/3Dmol.js
1
0
Fork 0
mirror of https://github.com/3dmol/3Dmol.js.git synced 2026-06-04 08:39:49 +09:00
Code Issues Packages Projects Releases Wiki Activity

Issue #678

Browse Source 
write_html in py3Dmol for turning viewer into webpage
This commit is contained in:
David Koes
2023-05-13 17:28:23 -04:00
parent fa5e8cb51f
commit 0153c320fd
1 changed files with 27 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
py3Dmol/py3Dmol/__init__.py
View File

@@ -1,6 +1,7 @@
import time
import json
import sys
import re
try:
import IPython.display
@@ -71,7 +72,7 @@ class view(object):
width = '%dpx'%width
if type(height) == int:
height = '%dpx'%height
self.startjs = '''<div id="%s" style="position: relative; width: %s; height: %s">
self.startjs = '''<div id="%s" style="position: relative; width: %s; height: %s;">
<p id="%s" style="background-color:#ffcccc;color:black">You appear to be running in JupyterLab (or JavaScript failed to load for some other reason). You need to install the 3dmol extension: <br>
<tt>jupyter labextension install jupyterlab_3dmol</tt></p>
</div>\n''' % (divid,width,height,warnid)
@@ -208,6 +209,31 @@ if(warn) {
self.updatejs = ''
return IPython.display.publish_display_data({'application/3dmoljs_load.v0':script, 'text/html': script},metadata={})
def write_html(self, f=None, fullpage=False):
'''Write html to reproduce viewer in a web page to a file.
f -- file name (str) or writeable file object; if unspecified html string is returned
fullpage -- instead of specified width/height make viewer fill the web page
'''
if f == None:
return self._make_html()
if type(f) == str:
f = open(f,'wt')
html = self._make_html()
html = f'''<html>
<body style="margin: 0; padding: 0; display: block;">
{html}
</body>
</html>'''
if fullpage:
html = re.sub(r'width: (\S+);', 'width: 100%;', html)
html = re.sub(r'height: (\S+);', 'height: 100vh;', html)
f.write(html)
@using_ipython
def png(self):
'''output png image of viewer, which must already be instantiated'''