I am not going to tell you how it can be done but inform you that there is information within the revit sdk for you to use and code it in python.
Search in sdk for the following:
revitLinkOptions - See get/set workset configuration
Worksetconfiguration - used to configure worksets for the above.
You will also need to apply these somehow to a existing Revit link instance.